Update main.py
Browse files
main.py
CHANGED
|
@@ -23,7 +23,7 @@ from utils import (
|
|
| 23 |
)
|
| 24 |
import torch
|
| 25 |
|
| 26 |
-
yolo_model = get_yolo_model(model_path='NewParser/best.pt')
|
| 27 |
#caption_model_processor = get_caption_model_processor(model_name="florence2", model_name_or_path="icon_caption_florence")
|
| 28 |
|
| 29 |
from ultralytics import YOLO
|
|
|
|
| 23 |
)
|
| 24 |
import torch
|
| 25 |
|
| 26 |
+
#yolo_model = get_yolo_model(model_path='NewParser/best.pt')
|
| 27 |
#caption_model_processor = get_caption_model_processor(model_name="florence2", model_name_or_path="icon_caption_florence")
|
| 28 |
|
| 29 |
from ultralytics import YOLO
|