Spaces:
Running
on
Zero
Running
on
Zero
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 |
|