Update app.py

app.py

@@ -487,8 +487,8 @@ def plot_human_pose(keypoints):
 
     # Close the plot to release resources
     plt.close()
-
-    return Image.open(buffer)
+    out = Image.open(buffer)
+    return out
 
 # Example usage:
 # keypoints = test_predict_plot.detach().numpy() # Replace with your 33 key points
@@ -518,12 +518,17 @@ def predict_pose(image):
 
   pred_keypoints = outputs.reshape(1,33,4)[0]
   pred_keypoints = pred_keypoints.detach().numpy()
-
   return plot_human_pose(pred_keypoints)
 
 # predict_pose(test_image)
-
-pose_detector = gr.Interface(predict_pose, gr.Image(type = 'pil') , "image")
+input_image = [
+    gr.Components.Image(type = 'pil'),
+]
+
+output_image = [
+    gr.Components.Image(type = 'pil'),
+]
+pose_detector = gr.Interface(fn = predict_pose, inputs = input_image , outputs = output_image )
 
 pose_detector.launch()