EmanHassan26 commited on
Commit
dcd81d4
·
verified ·
1 Parent(s): addc1a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)