Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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():
|