3v324v23 commited on
Commit
5e3ff34
·
1 Parent(s): 61b0e93

update checkpoint

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