visualization update

app.py

@@ -3,13 +3,13 @@ import numpy as np
 import gradio as gr
 from infer import detections
 
-
+'''
+'''
 import os
 os.system("mkdir data")
 os.system("mkdir data/models")
 os.system("wget https://www.cs.cmu.edu/~walt/models/walt_people.pth -O data/models/walt_people.pth")
 os.system("wget https://www.cs.cmu.edu/~walt/models/walt_vehicle.pth -O data/models/walt_vehicle.pth")
-
 def walt_demo(input_img):
     #detect_people = detections('configs/walt/walt_people.py', 'cuda:0', model_path='data/models/walt_people.pth')
     detect = detections('configs/walt/walt_vehicle.py', 'cuda:0', model_path='data/models/walt_vehicle.pth')
@@ -65,5 +65,6 @@ demo = gr.Interface(walt_demo,
 
 #demo.launch(server_name="0.0.0.0", server_port=7000)
 demo.launch()
+'''
 
 

requirements.txt

@@ -6,3 +6,5 @@ scikit-image
 imagesize
 torchvision==0.10.0
 imantics
+terminaltables
+pycocotools