amildravid4292 commited on
Commit
00275ee
·
verified ·
1 Parent(s): 0728566

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -395,7 +395,7 @@ def run_inversion(net, dict, pcs, epochs, weight_decay,lr):
395
  seed = 5
396
  cfg = 3.0
397
  steps = 25
398
- image = inference( prompt, negative_prompt, cfg, steps, seed)
399
  torch.save(network.proj, "model.pt" )
400
  return net, net, image
401
 
 
395
  seed = 5
396
  cfg = 3.0
397
  steps = 25
398
+ image = inference( net, prompt, negative_prompt, cfg, steps, seed)
399
  torch.save(network.proj, "model.pt" )
400
  return net, net, image
401