Upload folder using huggingface_hub

app.py

@@ -27,13 +27,15 @@ def postprocess_image(result: torch.Tensor, im_size: list) -> np.ndarray:
     return im_array
 
 
-def inference(orig_image):
+def inference(image):
 
     model_path = "ormbg.pth"
 
     net = ORMBG()
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
+    orig_image = Image.fromarray(image)
+
     if torch.cuda.is_available():
         net.load_state_dict(torch.load(model_path))
         net = net.cuda()