Spaces:
Runtime error
Runtime error
Hector Lopez
commited on
Commit
·
b237467
1
Parent(s):
d248ce1
Fixed import
Browse files
model.py
CHANGED
|
@@ -5,7 +5,9 @@ import torch
|
|
| 5 |
import numpy as np
|
| 6 |
import torchvision
|
| 7 |
|
| 8 |
-
|
|
|
|
|
|
|
| 9 |
|
| 10 |
def get_model(checkpoint_path):
|
| 11 |
extra_args = {}
|
|
|
|
| 5 |
import numpy as np
|
| 6 |
import torchvision
|
| 7 |
|
| 8 |
+
import icevision.models.ross.efficientdet
|
| 9 |
+
|
| 10 |
+
MODEL_TYPE = icevision.models.ross.efficientdet
|
| 11 |
|
| 12 |
def get_model(checkpoint_path):
|
| 13 |
extra_args = {}
|