Hugging Face's logo

Spaces:
PKaushik
/
humandetect
Build error

App Files Community
PKaushik commited on
Commit
c1a08cd
•
1 Parent(s): c5eaa2c

commit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import gradio as gr
8
 
9
  from inferer import Inferer
10
 
11
- pipeline = Inferer("PKaushik/humandetect", device='cuda')
12
  print(f"GPU on? {'🟢' if pipeline.device.type != 'cpu' else '🔴'}")
13
 
14
  def fn_image(image, conf_thres, iou_thres):
 
8
 
9
  from inferer import Inferer
10
 
11
+ pipeline = Inferer("nateraw/yolov6s", device='cuda')
12
  print(f"GPU on? {'🟢' if pipeline.device.type != 'cpu' else '🔴'}")
13
 
14
  def fn_image(image, conf_thres, iou_thres):