6Morpheus6 commited on
Commit
4d1343c
·
verified ·
1 Parent(s): 2cfde88
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -98,10 +98,10 @@ def infer(input_image, prompt, seed=42, randomize_seed=False, guidance_scale=2.5
98
 
99
  # @spaces.GPU
100
  def infer_example(input_image, prompt):
101
- image, seed, _ = infer(input_image, prompt)
102
  gc.collect()
103
  devicetorch.empty_cache(torch)
104
- return image, seed
105
 
106
  css="""
107
  #col-container {
 
98
 
99
  # @spaces.GPU
100
  def infer_example(input_image, prompt):
101
+ image, temp_file_path, seed, _ = infer(input_image, prompt)
102
  gc.collect()
103
  devicetorch.empty_cache(torch)
104
+ return image,temp_file_path, seed
105
 
106
  css="""
107
  #col-container {