Spaces:
Runtime error
Runtime error
update
Browse files
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 {
|