Update app.py

app.py

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