Update app.py

app.py

@@ -114,6 +114,8 @@ def pipe_inference(
     return all_outputs
 
 def resize_image(image, resolution):  
+    if not isinstance(image, np.ndarray):
+        image = np.array(image) 
     h, w = image.shape[:2]
     ratio = w/h
     if ratio > 1 :
@@ -122,10 +124,14 @@ def resize_image(image, resolution):
         resized_image = cv2.resize(image, (resolution, int(resolution/ratio)), interpolation=cv2.INTER_NEAREST)
     else:
         resized_image = cv2.resize(image, (resolution, resolution), interpolation=cv2.INTER_NEAREST)
-    return resized_image
+    
+    return Image.fromarray(resized_image)
     
     
 def preprocess_canny(image, resolution=128):
+    if not isinstance(image, np.ndarray):
+        image = np.array(image) 
+        
     processed_image = cv2.Canny(image, low_threshold, high_threshold)
     processed_image = processed_image[:, :, None]
     processed_image = np.concatenate([processed_image, processed_image, processed_image], axis=2)