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:
|