Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
·
3ed9715
1
Parent(s):
ccccb71
Update app.py
Browse files
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).
|