amildravid4292 commited on
Commit
2671fa3
·
verified ·
1 Parent(s): 4e3e10a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -93,7 +93,7 @@ def sample_then_run(network, unet):
93
  seed = 5
94
  cfg = 3.0
95
  steps = 25
96
- image = inference( prompt, negative_prompt, cfg, steps, seed)
97
  torch.save(network.proj, "model.pt" )
98
  #return
99
  return image, "model.pt", unet, network
 
93
  seed = 5
94
  cfg = 3.0
95
  steps = 25
96
+ image = inference( network, unet, prompt, negative_prompt, cfg, steps, seed)
97
  torch.save(network.proj, "model.pt" )
98
  #return
99
  return image, "model.pt", unet, network