Update app.py

app.py

@@ -7,10 +7,13 @@ os.system("pip3 install -r requirements.txt")
 os.system("pip3 install -v -e .")
 os.chdir("..")
 os.system("gdown https://drive.google.com/uc?id=1RE5vaKHYt_TL0cXNn6FmjO7aZx3sVfVA")
+from pathlib import Path
+
 
 def inference(video):
   os.system("python3 deploy/ONNXRuntime/onnx_inference.py -i "+video)
-  return "demo_output/"+video
+  vidname = Path(video).stem
+  return "demo_output/"+vidname+".mp4"
   
 
 title = "Anime2Sketch"