Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ def predict(image, prompt="high quality, best quality", negative_prompt="", guid
|
|
30 |
if negative_prompt == "":
|
31 |
negative_prompt = None
|
32 |
|
33 |
-
init_image =
|
34 |
init_image = init_image.resize((224, 224))
|
35 |
|
36 |
generator = torch.Generator(device="cpu").manual_seed(int(seed))
|
|
|
30 |
if negative_prompt == "":
|
31 |
negative_prompt = None
|
32 |
|
33 |
+
init_image = image.convert("RGB")
|
34 |
init_image = init_image.resize((224, 224))
|
35 |
|
36 |
generator = torch.Generator(device="cpu").manual_seed(int(seed))
|