Update app.py

app.py

@@ -508,8 +508,7 @@ model = SimpleCNN()
 model.load_state_dict(torch.load("model.pth"))
 model.eval()
 
-def predict_pose(path):
-  img = cv2.imread(str(path))
+def predict_pose(img):
   img= cv2.resize(img, (32,32))
   convert_tensor = transforms.ToTensor()
   tensor_img = convert_tensor(img)
@@ -523,11 +522,11 @@ def predict_pose(path):
 
 # predict_pose(test_image)
 input_image = [
-    gr.components.Image(type = "filepath"),
+    gr.components.Image(type = "pil"),
 ]
 
 output_image = [
-    gr.components.Image(type = "numpy"),
+    gr.components.Image(type = "pil"),
 ]
 pose_detector = gr.Interface(fn = predict_pose, inputs = input_image , outputs = output_image )