rgb

app.py

@@ -70,8 +70,7 @@ def predict(
 ):
     if input_image is None:
         raise gr.Error("Please upload an image.")
-    padded_image = pad_image(input_image).resize((1024, 1024))
-    padded_image.save(f"padded_image+{seed}.jpg")
+    padded_image = pad_image(input_image).resize((1024, 1024)).convert("RGB")
     image_lr = load_and_process_image(padded_image).to(device)
     generator = torch.manual_seed(seed)
     images = pipe(