Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -191,8 +191,8 @@ def generate_input_image(image_path,input_choice,seg_input_promt):
|
|
| 191 |
print(f"image_path: '{image_path}'")
|
| 192 |
print(f"input_choice: '{input_choice}'")
|
| 193 |
image_path = image_path[0]
|
| 194 |
-
gc.collect()
|
| 195 |
-
torch.cuda.empty_cache()
|
| 196 |
# ["Edge image","Segmented image","Depth Image"]
|
| 197 |
if input_choice == "Edge image" :
|
| 198 |
conditioning_image = generate_edgeImage(image_path)
|
|
|
|
| 191 |
print(f"image_path: '{image_path}'")
|
| 192 |
print(f"input_choice: '{input_choice}'")
|
| 193 |
image_path = image_path[0]
|
| 194 |
+
#gc.collect()
|
| 195 |
+
#torch.cuda.empty_cache()
|
| 196 |
# ["Edge image","Segmented image","Depth Image"]
|
| 197 |
if input_choice == "Edge image" :
|
| 198 |
conditioning_image = generate_edgeImage(image_path)
|