Update main.py

main.py

@@ -36,7 +36,7 @@ from ultralytics import YOLO
 #except:
     #yolo_model = YOLO("best.pt")
 from transformers import AutoProcessor, AutoModelForCausalLM
-
+from transformers import AutoProcessor, Blip2ForConditionalGeneration
 # Correctly load the processor and model for Blip-2
 try:
     processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base", trust_remote_code=True)