Object detection app
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ from timm.data import resolve_data_config
|
|
13 |
from timm.data.transforms_factory import create_transform
|
14 |
|
15 |
from model import Model
|
16 |
-
import
|
17 |
|
18 |
|
19 |
# Use GPU if available
|
|
|
13 |
from timm.data.transforms_factory import create_transform
|
14 |
|
15 |
from model import Model
|
16 |
+
from get_output import visualize_output
|
17 |
|
18 |
|
19 |
# Use GPU if available
|