Update multimodal_ai.py

multimodal_ai.py

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