Update app.py
Browse files
app.py
CHANGED
|
@@ -32,7 +32,6 @@ detector50 = pipeline(model="facebook/detr-resnet-50")
|
|
| 32 |
detector101 = pipeline(model="facebook/detr-resnet-101")
|
| 33 |
|
| 34 |
if torch.cuda.is_available():
|
| 35 |
-
print("use cuda")
|
| 36 |
detector50.model.to('cuda')
|
| 37 |
detector101.model.to('cuda')
|
| 38 |
|
|
|
|
| 32 |
detector101 = pipeline(model="facebook/detr-resnet-101")
|
| 33 |
|
| 34 |
if torch.cuda.is_available():
|
|
|
|
| 35 |
detector50.model.to('cuda')
|
| 36 |
detector101.model.to('cuda')
|
| 37 |
|