convert H264

app.py

@@ -14,7 +14,7 @@ from Modules.PoseEstimation.pose_agent import agent_executor
 
 mistral_api_key = os.getenv("MISTRAL_API_KEY")
 from Modules.PoseEstimation import pose_estimator
-from utils import save_uploaded_file
+from utils import save_uploaded_file, encode_video_H264
 
 
 st.set_page_config(layout="wide", initial_sidebar_state="collapsed")
@@ -97,8 +97,9 @@ with col2:
                 predict_list.sort()
                 predict_dir = predict_list[-1]
                 file_name = os.listdir(os.path.join('/home/user/.pyenv/runs', 'pose', predict_dir))[0]
-                print(file_name)
-                st.video(os.path.join('/home/user/.pyenv/runs', 'pose', predict_dir, file_name), loop=True)
+                file_path =os.path.join('/home/user/.pyenv/runs', 'pose', predict_dir, file_name)
+                file_path = encode_video_H264(file_path, remove_original=True)
+                st.video(file_path, loop=True)
             else :
                 st.video(video_uploaded)