Update test_ai_integration.py

test_ai_integration.py

@@ -158,21 +158,21 @@ def test_ai_integration():
             })
 
         # Load BLIP-2 Large model directly to WebSocket storage
-        from transformers import AutoModelForCausalLM, AutoProcessor
-        model_id = "microsoft/florence-2-large"
+        from transformers import Blip2ForConditionalGeneration, Blip2Processor
+        model_id = "Salesforce/blip2-opt-2.7b"
         print(f"Loading model {model_id} directly to WebSocket storage...")
         
         try:
             # Load model and processor with proper error handling
-            model = AutoModelForCausalLM.from_pretrained(
-                model_id, 
+            model = Blip2ForConditionalGeneration.from_pretrained(
+                model_id,
                 trust_remote_code=True,
                 device_map="auto",  # Allow automatic device mapping
                 torch_dtype="auto"   # Use appropriate dtype
             )
             
-            processor = AutoProcessor.from_pretrained(
-                model_id, 
+            processor = Blip2Processor.from_pretrained(
+                model_id,
                 trust_remote_code=True
             )