Update app.py
Browse files
app.py
CHANGED
|
@@ -32,6 +32,7 @@ detector50 = pipeline(model="facebook/detr-resnet-50")
|
|
| 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 |
|
|
|
|
| 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 |
|