cuda

Dockerfile

@@ -4,7 +4,7 @@ WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 
 #RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
-RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/cu111 
+RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
 #https://download.pytorch.org/whl/torch_stable.ht
 RUN pip install --no-cache-dir --upgrade -U openmim
 RUN mim install --no-cache-dir --upgrade mmengine

main.py

@@ -179,19 +179,21 @@ def pose2dhand(video, kpt_threshold):
 
 def run_UI():
     with gr.Blocks() as demo:
-        with gr.Column():            
+        with gr.Column():       
             with gr.Tab("Upload video"):
-                with gr.Row():
-                    with gr.Column():
-                        video_input = gr.Video(source="upload", type="filepath", height=612)
-                        # Insert slider with kpt_thr
-                        file_kpthr = gr.Slider(minimum=0.1, maximum=1, step=20, default=0.3, label='Keypoint threshold')
-
-                        submit_pose_file = gr.Button("Make 2d pose estimation")
-                        submit_pose3d_file = gr.Button("Make 3d pose estimation")
-                        submit_hand_file = gr.Button("Make 2d hand estimation")
-                        submit_detect_file = gr.Button("Detect and track objects")
-                    with gr.Column():
+                with gr.Column():
+                    with gr.Row():
+                        with gr.Column():
+                            video_input = gr.Video(source="upload", type="filepath", height=612)
+                            # Insert slider with kpt_thr
+                            file_kpthr = gr.Slider(minimum=0.1, maximum=1, step=20, default=0.3, label='Keypoint threshold')
+
+                            submit_pose_file = gr.Button("Make 2d pose estimation")
+                            submit_pose3d_file = gr.Button("Make 3d pose estimation")
+                            submit_hand_file = gr.Button("Make 2d hand estimation")
+                            submit_detect_file = gr.Button("Detect and track objects")
+                            
+                    with gr.Row():
                         video_output1 = gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True)
                         video_output2 = gr.PlayableVideo(height=512,  label = "Estimate human 3d poses", show_label=True)
                         video_output3 = gr.PlayableVideo(height=512,  label = "Estimate human hand poses", show_label=True)