Update app.py
Browse files
app.py
CHANGED
|
@@ -327,7 +327,7 @@ class main():
|
|
| 327 |
seed = 5
|
| 328 |
cfg = 3.0
|
| 329 |
steps = 25
|
| 330 |
-
image = self.inference(
|
| 331 |
torch.save(self.weights.cpu().detach(), "model.pt" )
|
| 332 |
return image, "model.pt"
|
| 333 |
|
|
|
|
| 327 |
seed = 5
|
| 328 |
cfg = 3.0
|
| 329 |
steps = 25
|
| 330 |
+
image = self.inference(prompt, negative_prompt, cfg, steps, seed)
|
| 331 |
torch.save(self.weights.cpu().detach(), "model.pt" )
|
| 332 |
return image, "model.pt"
|
| 333 |
|