Ahsen Khaliq commited on
Commit
8f204e9
·
1 Parent(s): a911b9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -11,9 +11,10 @@ from pathlib import Path
11
 
12
 
13
  def inference(video):
14
- os.system("python3 deploy/ONNXRuntime/onnx_inference.py -i "+video)
15
- vidname = Path(video).stem
16
- return "demo_output/"+vidname+".mp4"
 
17
 
18
 
19
  title = "Anime2Sketch"
 
11
 
12
 
13
  def inference(video):
14
+ oldext = os.path.splitext(video)[1]
15
+ os.rename(video, "test" + oldext)
16
+ os.system("python3 deploy/ONNXRuntime/onnx_inference.py -i test"+oldtext)
17
+ return "demo_output/test"+oldtext
18
 
19
 
20
  title = "Anime2Sketch"