Update multimodal_ai.py

multimodal_ai.py

@@ -17,7 +17,8 @@ class MultimodalAI:
         login(token=self.HUGGINGFACE_TOKEN)
 
         # Model selection
-        self.model_name = "openai-community/gpt2"
+        #self.model_name = "openai-community/gpt2"
+        self.model_name = "liuhaotian/llava-v1.6-34b"
 
         # Check if a CUDA-enabled GPU is available.
         # If available, move the model to the GPU (cuda:0) for faster computation.