Update app.py

app.py

@@ -444,6 +444,8 @@ model = SimpleCNN()
 # keypoints = test_actual_plot  # Replace with your 33 key points
 # plot_human_pose(keypoints)
 
+from io import BytesIO
+
 def plot_human_pose(keypoints):
     # Create a figure and axis
     fig, ax = plt.subplots()
@@ -477,7 +479,15 @@ def plot_human_pose(keypoints):
     ax.set_aspect('equal', adjustable='datalim')
     plt.title('Predicted Pose')
     plt.axis('off')
-    plt.show()
+    
+    buffer = BytesIO()
+    plt.savefig(buffer, format="png")
+    buffer.seek(0)  # Reset the buffer position to the beginning
+
+    # Close the plot to release resources
+    plt.close()
+
+    return buffer
 
 # Example usage:
 # keypoints = test_predict_plot.detach().numpy() # Replace with your 33 key points
@@ -512,7 +522,7 @@ def predict_pose(image):
 
 # predict_pose(test_image)
 
-pose_detector = gr.Interface(fn = predict_pose, inputs = [gr.Components.Image(type='pil', label = 'input')], outputs = [gr.Components.Image(type='numpy', label = 'output')])
+pose_detector = gr.Interface(predict_pose, gr.Image(type = 'pil') , "image")
 
 pose_detector.launch()