Linoy Tsaban commited on
Commit
7f54407
1 Parent(s): e496c75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -101,14 +101,11 @@ def edit(tar_prompt="",
101
  tar_cfg_scale=15,
102
 
103
  ):
104
- outputs = []
105
- num_generations = 1
106
- for i in range(num_generations):
107
- out = sample(wt=inversion_map['wt'], zs= inversion_map['zs'], wts=inversion_map['wts'], prompt_tar=tar_prompt,
108
- cfg_scale_tar=tar_cfg_scale, skip=skip)
109
- outputs.append(out)
110
 
111
- return outputs
112
 
113
  def reset():
114
  inversion_map.clear()
 
101
  tar_cfg_scale=15,
102
 
103
  ):
104
+
105
+ out = sample(wt=inversion_map['wt'], zs= inversion_map['zs'], wts=inversion_map['wts'], prompt_tar=tar_prompt,
106
+ cfg_scale_tar=tar_cfg_scale, skip=skip)
 
 
 
107
 
108
+ return out
109
 
110
  def reset():
111
  inversion_map.clear()