Update app.py

app.py

@@ -10,10 +10,14 @@ model = torch.hub.load("PeterL1n/RobustVideoMatting", "mobilenetv3") # or "resne
 convert_video = torch.hub.load("PeterL1n/RobustVideoMatting", "converter")
 
 def inference(video):
-  os.system('rm output.mp4')
+  #os.system('rm output.mp4')
   #clip = VideoFileClip(video).subclip(0, 5)
   #clip.write_videofile("output.mp4")
-  os.system('ffmpeg -ss 00:00:00 -i '+ video +' -to 00:00:05 -c copy -y output.mp4')
+  #os.system('ffmpeg -ss 00:00:00 -i '+ video +' -to 00:00:05 -c copy -y output.mp4')
+  clip = VideoFileClip(video)
+  print(clip.duration)
+  if clip.duration > 5:
+      return 'trim.mp4',"trim.mp4","trim.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.