Update app.py

app.py

@@ -8,9 +8,10 @@ model = torch.hub.load("PeterL1n/RobustVideoMatting", "mobilenetv3") # or "resne
 convert_video = torch.hub.load("PeterL1n/RobustVideoMatting", "converter")
 
 def inference(video):
+  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=video,        # A video file or an image sequence directory.
+      input_source='output.mp4',        # A video file or an image sequence directory.
       input_resize=(500, 500),       # [Optional] Resize the input (also the output).
       downsample_ratio=0.25,           # [Optional] If None, make downsampled max size be 512px.
       output_type='video',             # Choose "video" or "png_sequence"