Spaces:

Baptlem
/

UCDR-Net

Runtime error

Baptlem commited on May 2, 2023

Commit

5461399

1 Parent(s): 68696f0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -114,8 +114,7 @@ def pipe_inference(
     return all_outputs
 def resize_image(image, resolution):
-    print(image.shape)
-    h, w = image.shape
     ratio = w/h
     if ratio > 1 :
         resized_image = cv2.resize(image, (int(resolution*ratio), resolution), interpolation=cv2.INTER_NEAREST)

     return all_outputs
 def resize_image(image, resolution):
+    h, w = image.shape[:2]
     ratio = w/h
     if ratio > 1 :
         resized_image = cv2.resize(image, (int(resolution*ratio), resolution), interpolation=cv2.INTER_NEAREST)