cuda

Dockerfile

@@ -6,6 +6,8 @@ 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/torch_stable.ht -extra-index-url https://download.pytorch.org/whl/cu111
 #https://download.pytorch.org/whl/torch_stable.ht
+RUN pip uninstall torch -y
+RUN pip uninstall torchvision -y
 
 RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu113
 # --index-url https://download.pytorch.org/whl/cu117

main.py

@@ -189,10 +189,10 @@ def run_UI():
                             # 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")
+                            submit_pose_file = gr.Button("Make 2d pose estimation", variant="primary")
+                            submit_pose3d_file = gr.Button("Make 3d pose estimation", variant="primary")
+                            submit_hand_file = gr.Button("Make 2d hand estimation", variant="primary")
+                            submit_detect_file = gr.Button("Detect and track objects", variant="primary")
 
                     with gr.Row():
                         video_output1 = gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True)
@@ -209,10 +209,10 @@ def run_UI():
                             
                             web_kpthr = gr.Slider(minimum=0.1, maximum=1, step=20, default=0.3, label='Keypoint threshold')
 
-                            submit_pose_web = gr.Button("Make 2d pose estimation")
-                            submit_pose3d_web = gr.Button("Make 3d pose estimation")
-                            submit_hand_web = gr.Button("Make 2d hand estimation")
-                            submit_detect_web = gr.Button("Detect and track objects")
+                            submit_pose_web = gr.Button("Make 2d pose estimation", variant="primary")
+                            submit_pose3d_web = gr.Button("Make 3d pose estimation", variant="primary")
+                            submit_hand_web = gr.Button("Make 2d hand estimation", variant="primary")
+                            submit_detect_web = gr.Button("Detect and track objects", variant="primary")
                     with gr.Row():
                         webcam_output1 = gr.PlayableVideo(height=512,  label = "Estimate human 2d poses", show_label=True)
                         webcam_output2 = gr.PlayableVideo(height=512,  label = "Estimate human 3d poses", show_label=True)

requirements.txt

@@ -3,4 +3,6 @@ numpy
 opencv-python
 ultralytics
 lap
-MoviePy
+MoviePy
+torch
+torchvision