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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,13 +2,14 @@ import numpy as np
2
  import torch
3
  import gradio as gr
4
  from infer import detections
5
- '''
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
  def walt_demo(input_img, confidence_threshold):
13
  #detect_people = detections('configs/walt/walt_people.py', 'cuda:0', model_path='data/models/walt_people.pth')
14
  if torch.cuda.is_available() == False:
 
2
  import torch
3
  import gradio as gr
4
  from infer import detections
5
+
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: