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:
|