Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import gradio as gr
+from uuid import uuid4
 from pipelines.pipeline import InferencePipeline
 
 TITLE = """
@@ -64,6 +65,8 @@ CSS = """
     }
 """
 
+FFMPEG_COMMAND = "-loglevel error -y -r 25 -pix_fmt yuv420p -f mp4"
+
 pipelines = {
     "VSR(mediapipe)": InferencePipeline("./configs/LRS3_V_WER19.1.ini", device="cpu", face_track=True, detector="mediapipe"),
     "ASR": InferencePipeline("./configs/LRS3_A_WER1.0.ini", device="cpu", face_track=True, detector="mediapipe"),
@@ -71,9 +74,15 @@ pipelines = {
 }
 
 def fn(pipeline_type, filename):
+    directory = "./tmp"
+    if not os.path.exists(directory):
+        os.makedirs(directory)
+    dst_filename = os.path.join(directory, str(uuid4())[:8]+".mp4")
+    command_string = f"ffmpeg -i {filename} {FFMPEG_COMMAND} {dst_filename}"
+    os.system(command_string)
     selected_pipeline_instance = pipelines[pipeline_type]
-    landmarks = selected_pipeline_instance.process_landmarks(filename, landmarks_filename=None)
-    data = selected_pipeline_instance.dataloader.load_data(filename, landmarks)
+    landmarks = selected_pipeline_instance.process_landmarks(dst_filename, landmarks_filename=None)
+    data = selected_pipeline_instance.dataloader.load_data(dst_filename, landmarks)
     transcript = selected_pipeline_instance.model.infer(data)
     return transcript