update video codec

app.py

@@ -16,6 +16,7 @@ import gradio as gr
 import numpy as np
 import pandas as pd
 import plotly.graph_objs as go
+import spaces
 import torch
 from metrabs_pytorch.scripts.run_video import run_metrabs_video
 from myo_tools.mjs.marker.marker_api import get_marker_names
@@ -46,7 +47,7 @@ def draw_keypoints(frame, poses2d, radius=10):
 
 
 def save_video_with_keypoints(results, output_video):
-    fourcc = cv2.VideoWriter_fourcc(*"avc1")
+    fourcc = cv2.VideoWriter_fourcc(*"mp4v")
     out = cv2.VideoWriter(
         output_video,
         fourcc,
@@ -222,6 +223,7 @@ def run_retargeting_c3d(api_key, c3d_files, markerset_file):
         )
 
 
+@spaces.GPU
 def run_retargeting_video(
     api_key,
     video_file="",
@@ -307,7 +309,7 @@ def run_retargeting_video(
             visible=False,
         ), video_with_keypoints
         client = Client(api_key=api_key)
-
+        print(f"🔹 MyoSDK client initialized in { time.time() - init_time:.2f} seconds")
         status.append(
             f"🔹 MyoSDK client initialized in { time.time() - init_time:.2f} seconds"
         )
@@ -325,6 +327,7 @@ def run_retargeting_video(
         status.append(
             f"🔹 Markerset file uploaded in {time.time() - init_time:.2f} seconds"
         )
+        print(f"🔹 Markerset file uploaded in {time.time() - init_time:.2f} seconds")
 
         yield "\n".join(status), None, None, gr.update(value=[]), gr.update(
             visible=False
@@ -381,7 +384,8 @@ def run_retargeting_video(
         data = np.load(out_path)
         joint_angles = data["joint_angles_degrees"].squeeze()
         joint_names = data["joint_names"]
-
+        print(joint_angles)
+        print(joint_names)
         df = pd.DataFrame(joint_angles, columns=[jn for jn in joint_names])
         df.insert(0, "frame", df.index)