Update main.py

main.py

@@ -23,7 +23,7 @@ from utils import (
 )
 import torch
 
-yolo_model = get_yolo_model(model_path='best.pt')
+#yolo_model = get_yolo_model(model_path='best.pt')
 #caption_model_processor = get_caption_model_processor(model_name="florence2", model_name_or_path="icon_caption_florence")
 
 from ultralytics import YOLO
@@ -31,12 +31,12 @@ from ultralytics import YOLO
 #if not os.path.exists("NewParser/best.pt"):
     #os.makedirs("NewParser/best.pt")
 
-try:
-    yolo_model = YOLO("best.pt").to("cuda")
-except:
-    yolo_model = YOLO("best.pt")
+#try:
+    #yolo_model = YOLO("best.pt").to("cuda")
+#except:
+    #yolo_model = YOLO("best.pt")
 
-from transformers import AutoProcessor, AutoModelForCausalLM
+#from transformers import AutoProcessor, AutoModelForCausalLM
 
 processor = AutoProcessor.from_pretrained(
     "microsoft/Florence-2-base", trust_remote_code=True