amildravid4292 commited on
Commit
38f9244
·
verified ·
1 Parent(s): 8091383

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -327,7 +327,7 @@ class main():
327
  seed = 5
328
  cfg = 3.0
329
  steps = 25
330
- image = self.inference( weights, prompt, negative_prompt, cfg, steps, seed)
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