dineshreddy commited on
Commit
bf94afb
·
1 Parent(s): a9a719b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,10 +6,10 @@ from infer import detections
6
  import os
7
  os.system("mkdir data")
8
  os.system("mkdir data/models")
 
9
  os.system("wget https://www.cs.cmu.edu/~walt/models/walt_people.pth -O data/models/walt_people.pth")
10
  os.system("wget https://www.cs.cmu.edu/~walt/models/walt_vehicle.pth -O data/models/walt_vehicle.pth")
11
  '''
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:
 
6
  import os
7
  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: