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

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, input_image["background"])
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():