dineshreddy commited on
Commit
b6aeebb
1 Parent(s): bf94afb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,8 +8,8 @@ os.system("mkdir data")
8
  os.system("mkdir data/models")
9
  '''
10
  os.system("wget https://www.cs.cmu.edu/~walt/models/walt_people.pth -O data/models/walt_people.pth")
11
- os.system("wget https://www.cs.cmu.edu/~walt/models/walt_vehicle.pth -O data/models/walt_vehicle.pth")
12
  '''
 
13
  def walt_demo(input_img, confidence_threshold):
14
  #detect_people = detections('configs/walt/walt_people.py', 'cuda:0', model_path='data/models/walt_people.pth')
15
  if torch.cuda.is_available() == False:
 
8
  os.system("mkdir data/models")
9
  '''
10
  os.system("wget https://www.cs.cmu.edu/~walt/models/walt_people.pth -O data/models/walt_people.pth")
 
11
  '''
12
+ os.system("wget https://www.cs.cmu.edu/~walt/models/walt_vehicle.pth -O data/models/walt_vehicle.pth")
13
  def walt_demo(input_img, confidence_threshold):
14
  #detect_people = detections('configs/walt/walt_people.py', 'cuda:0', model_path='data/models/walt_people.pth')
15
  if torch.cuda.is_available() == False: