Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|
|
|
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:
|