Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -35,7 +35,7 @@ from ultralytics import YOLO
|
|
35 |
#yolo_model = YOLO("best.pt").to("cuda")
|
36 |
#except:
|
37 |
#yolo_model = YOLO("best.pt")
|
38 |
-
|
39 |
|
40 |
# Correctly load the processor and model for Blip-2
|
41 |
try:
|
|
|
35 |
#yolo_model = YOLO("best.pt").to("cuda")
|
36 |
#except:
|
37 |
#yolo_model = YOLO("best.pt")
|
38 |
+
from transformers import AutoProcessor, AutoModelForCausalLM
|
39 |
|
40 |
# Correctly load the processor and model for Blip-2
|
41 |
try:
|