Update app.py

app.py

@@ -20,10 +20,12 @@ def save_json(candidate, subset, json_file_path='./pose_data.json'):
     return json_file_path
 
 def pose_estimation(test_image):
-    bgr_image_path = './test.png'
-    with open(bgr_image_path, 'wb') as bgr_file:
-        bgr_file.write(test_image)
-    # Load the estimation models
+    oriImg = cv2.cvtColor(oriImg, cv2.COLOR_RGB2BGR)
+    
+    # bgr_image_path = './test.png'
+    # with open(bgr_image_path, 'wb') as bgr_file:
+    #     bgr_file.write(test_image)
+    # # Load the estimation models
     body_estimation = Body('model/body_pose_model.pth')
     hand_estimation = Hand('model/hand_pose_model.pth')
 
@@ -64,7 +66,8 @@ def convert_image_to_bytes(image_path):
 with gr.Blocks() as demo:
     gr.Markdown("# Pose Estimation")
     with gr.Row():
-        image = gr.File(label="Upload Image", type="binary")
+        # image = gr.File(label="Upload Image", type="binary")
+        image = gr.Image(label="Upload Image", type="numpy")
         output_image = gr.Image(label="Estimation Result")
         output_json = gr.File(label="Download Pose Data as JSON", type="filepath")  # Add JSON output
     submit_button = gr.Button("Start Estimation")