Update app.py

app.py

@@ -4,10 +4,11 @@ os.system("pip3 install cython_bbox gdown 'git+https://github.com/cocodataset/co
 os.system("pip3 install -v -e .")
 os.system("gdown https://drive.google.com/uc?id=1RE5vaKHYt_TL0cXNn6FmjO7aZx3sVfVA")
 
-def inference(video):
-  os.system("python onnx_inference.py -i "+video)
-  os.system("ls .")
-  return "*.mp4"
+def inference(image):
+  os.system("python3 tools/demo_track.py image -f exps/example/mot/yolox_x_mix_det.py -c ./bytetrack_x_mot17.pth.tar --save_result --device 'cpu' --path "+image.name)
+  #os.system("python onnx_inference.py -i "+video)
+  #os.system("ls .")
+  return image.name
   
 
 title = "Anime2Sketch"
@@ -16,8 +17,8 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2104.057
 
 gr.Interface(
     inference, 
-    "video", 
-    "playable_video",
+    gr.inputs.Image(type="file"), 
+    gr.outputs.Image(type="file"),
     title=title,
     description=description,
     article=article,