Update detection_utils.py (#7)
Browse files- Update detection_utils.py (756866a5084308b93d99a9adda5aa778a991e612)
Co-authored-by: Cyril Achard <C-Achard@users.noreply.huggingface.co>
- detection_utils.py +3 -1
detection_utils.py
CHANGED
|
@@ -35,7 +35,9 @@ def predict_md(im,
|
|
| 35 |
'custom', #model
|
| 36 |
megadetector_model, # args for callable model
|
| 37 |
force_reload=True,
|
| 38 |
-
device=md_device
|
|
|
|
|
|
|
| 39 |
|
| 40 |
# send model to gpu if possible
|
| 41 |
if (md_device == torch.device('cuda')):
|
|
|
|
| 35 |
'custom', #model
|
| 36 |
megadetector_model, # args for callable model
|
| 37 |
force_reload=True,
|
| 38 |
+
device=md_device,
|
| 39 |
+
trust_repo=True
|
| 40 |
+
)
|
| 41 |
|
| 42 |
# send model to gpu if possible
|
| 43 |
if (md_device == torch.device('cuda')):
|