Update app.py

app.py

@@ -1,16 +1,31 @@
-import os, torchvision, transformers
+import os, torchvision, transformers, subprocess
 from functools import partial
 import gradio as gr
 
-from data.utils import ffmpeg_once
-
-from .inference import LiveInfer
+from inference import LiveInfer
 logger = transformers.logging.get_logger('liveinfer')
 
 # python -m demo.app --resume_from_checkpoint ... 
 
 liveinfer = LiveInfer()
 
+def ffmpeg_once(src_path: str, dst_path: str, *, fps: int = None, resolution: int = None, pad: str = '#000000', mode='bicubic'):
+    os.makedirs(os.path.dirname(dst_path), exist_ok=True)
+    command = [
+        './ffmpeg/ffmpeg',
+        '-y',
+        '-sws_flags', mode,
+        '-i', src_path,
+        '-an',
+        '-threads', '10',
+    ]
+    if fps is not None:
+        command += ['-r', str(fps)]
+    if resolution is not None:
+        command += ['-vf', f"scale='if(gt(iw\\,ih)\\,{resolution}\\,-2)':'if(gt(iw\\,ih)\\,-2\\,{resolution})',pad={resolution}:{resolution}:(ow-iw)/2:(oh-ih)/2:color='{pad}'"]
+    command += [dst_path]
+    subprocess.run(command, check=True)
+    
 css = """
     #gr_title {text-align: center;}
     #gr_video {max-height: 480px;}