Ahsen Khaliq commited on
Commit
3ed9715
·
1 Parent(s): ccccb71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -8,6 +8,7 @@ model = torch.hub.load("PeterL1n/RobustVideoMatting", "mobilenetv3") # or "resne
8
  convert_video = torch.hub.load("PeterL1n/RobustVideoMatting", "converter")
9
 
10
  def inference(video):
 
11
  os.system('ffmpeg -ss 00:00:00 -i '+ video +' -to 00:00:05 -c copy output.mp4')
12
  convert_video(
13
  model, # The loaded model, can be on any device (cpu or cuda).
 
8
  convert_video = torch.hub.load("PeterL1n/RobustVideoMatting", "converter")
9
 
10
  def inference(video):
11
+ os.system('rm output.mp4')
12
  os.system('ffmpeg -ss 00:00:00 -i '+ video +' -to 00:00:05 -c copy output.mp4')
13
  convert_video(
14
  model, # The loaded model, can be on any device (cpu or cuda).