Update app.py

app.py

@@ -3,13 +3,18 @@ os.system('pip install gradio --upgrade')
 os.system('pip freeze')
 import torch
 import gradio as gr
+from moviepy.editor import *
+
 model = torch.hub.load("PeterL1n/RobustVideoMatting", "mobilenetv3") # or "resnet50"
 
 convert_video = torch.hub.load("PeterL1n/RobustVideoMatting", "converter")
 
 def inference(video):
-  os.system('rm output.mp4')
-  os.system('ffmpeg -ss 00:00:00 -i '+ video +' -to 00:00:05 -c copy output.mp4')
+ os.system('rm output.mp4')
+ clip = VideoFileClip(video).subclip(0, 5)
+ clip.write_videofile("output.mp4")
+ # os.system('rm output.mp4')
+ # os.system('ffmpeg -ss 00:00:00 -i '+ video +' -to 00:00:05 -c copy output.mp4')
   convert_video(
       model,                           # The loaded model, can be on any device (cpu or cuda).
       input_source='output.mp4',        # A video file or an image sequence directory.