yrosenbloom commited on
Commit
8856f36
·
verified ·
1 Parent(s): bea163c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ depth_model.to(device)
22
 
23
  def process_image(image):
24
  # Ensure image is in RGB format and resize
25
- image = ImageOps.exif_transpose(image).resize((512, 512))
26
 
27
  # Perform segmentation
28
  inputs = seg_feature_extractor(images=image, return_tensors="pt").to(device)
 
22
 
23
  def process_image(image):
24
  # Ensure image is in RGB format and resize
25
+ image = image.resize((512, 512))
26
 
27
  # Perform segmentation
28
  inputs = seg_feature_extractor(images=image, return_tensors="pt").to(device)