animrods commited on
Commit
8080c6c
·
verified ·
1 Parent(s): ba04723

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = dict["image"].convert("RGB")
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))