anzorq commited on
Commit
9c52fdd
1 Parent(s): d71d9d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -41,8 +41,10 @@ if torch.cuda.is_available():
41
 
42
  device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
43
 
44
- def inference(model, prompt, img, strength, guidance, steps, seed):
 
45
  generator = torch.manual_seed(seed) if seed != 0 else None
 
46
  if img is not None:
47
  return img_inference(model, prompt, img, strength, guidance, steps, generator)
48
  else:
 
41
 
42
  device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"
43
 
44
+ def inference(model, img, strength, prompt, guidance, steps, seed):
45
+
46
  generator = torch.manual_seed(seed) if seed != 0 else None
47
+
48
  if img is not None:
49
  return img_inference(model, prompt, img, strength, guidance, steps, generator)
50
  else: