amildravid4292 commited on
Commit
bfda829
·
verified ·
1 Parent(s): 1f40b0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -195,7 +195,7 @@ def edit_inference(input_image, prompt, negative_prompt, guidance_scale, ddim_st
195
  network.proj = torch.nn.Parameter(original_weights)
196
  network.reset()
197
 
198
- return image
199
 
200
  # @spaces.GPU()
201
  def sample_then_run():
 
195
  network.proj = torch.nn.Parameter(original_weights)
196
  network.reset()
197
 
198
+ return [image]
199
 
200
  # @spaces.GPU()
201
  def sample_then_run():