foryahasake commited on
Commit
d44f300
·
verified ·
1 Parent(s): 6780fc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def inference(image_url, image, min_score):
49
  cfg.MODEL.DEVICE = "cpu"
50
  else:
51
  cfg.MODEL.DEVICE = "cuda"
52
- print( cfg.MODEL.DEVICE)
53
  if image_url:
54
  r = requests.get(image_url)
55
  if r:
 
49
  cfg.MODEL.DEVICE = "cpu"
50
  else:
51
  cfg.MODEL.DEVICE = "cuda"
52
+ print( cfg.MODEL.DEVICE)
53
  if image_url:
54
  r = requests.get(image_url)
55
  if r: