Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
| 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:
|