Upload folder using huggingface_hub

app.py

@@ -44,10 +44,10 @@ def inference(image):
     net.eval()
 
     model_input_size = [1024, 1024]
-    orig_im_size = orig_image.shape[0:2]
-    image = preprocess_image(orig_image, model_input_size).to(device)
+    orig_im_size = orig_image.size
+    processed_image = preprocess_image(orig_image, model_input_size).to(device)
 
-    result = net(image)
+    result = net(processed_image)
 
     # post process
     result_image = postprocess_image(result[0][0], orig_im_size)