Update main.py
Browse files
main.py
CHANGED
|
@@ -29,7 +29,7 @@ caption_model_processor = get_caption_model_processor(model_name="florence2", mo
|
|
| 29 |
from ultralytics import YOLO
|
| 30 |
|
| 31 |
if not os.path.exists("/data/icon_detect"):
|
| 32 |
-
os.makedirs("/data/icon_detect")
|
| 33 |
|
| 34 |
try:
|
| 35 |
yolo_model = YOLO("NewParser/best.pt").to("cuda")
|
|
|
|
| 29 |
from ultralytics import YOLO
|
| 30 |
|
| 31 |
if not os.path.exists("/data/icon_detect"):
|
| 32 |
+
os.makedirs("/data/icon_detect")
|
| 33 |
|
| 34 |
try:
|
| 35 |
yolo_model = YOLO("NewParser/best.pt").to("cuda")
|