animrods commited on
Commit
e55b6fb
·
verified ·
1 Parent(s): 3d2512c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def predict(dict, prompt="high quality, best quality", negative_prompt="", guida
33
  init_image = dict["image"].convert("RGB")
34
  init_image = init_image.resize((224, 224))
35
 
36
- generator = torch.Generator(device="cpu").manual_seed(seed)
37
 
38
  output = pipe(
39
  prompt=prompt,
 
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))
37
 
38
  output = pipe(
39
  prompt=prompt,