new config

main_noweb.py

@@ -15,6 +15,7 @@ import moviepy.editor as moviepy
 import os
 import glob
 import uuid
+import json
 
 # Image manipulation
 import numpy as np
@@ -30,7 +31,7 @@ else:
 os.system("nvidia-smi")
 
 print("[INFO]: Imported modules!")
-human = MMPoseInferencer(pose2d="human")
+human = MMPoseInferencer("dekr_hrnet-w32_8xb10-140e_coco-512x512.py")
 hand = MMPoseInferencer("hand")
 human3d = MMPoseInferencer(pose3d="human3d")
 track_model = YOLO('yolov8n.pt')  # Load an official Detect model
@@ -161,7 +162,10 @@ def pose2d(video, kpt_threshold):
     print(glob.glob(os.path.join(add_dir, "*.json")))
     print(os.listdir(add_dir))
 
-    return "".join(out_file), "".join(kpoints)
+    with open('path_to_file/person.json', 'r') as f:
+        data = json.load(kpoints)
+
+    return "".join(out_file), data # "".join(kpoints)
 
 
 def pose2dhand(video, kpt_threshold):
@@ -196,7 +200,7 @@ with block:
             with gr.Column():
                 with gr.Row():
                     with gr.Column():
-                        video_input = gr.Video(source="upload", type="filepath", height=612)
+                        video_input = gr.Video(source="upload", type="filepath", height=512, width=512)
                         # Insert slider with kpt_thr
                         file_kpthr = gr.Slider(0, 1, value=0.3, label='Keypoint threshold')
                         with gr.Row():