Update LLM.py

LLM.py

@@ -24,8 +24,11 @@ class Bot():
                           "google/gemma-1.1-7b-it"]
         self.google_models = ["gemini-1.0-pro", "gemini-1.5-flash",\
                               "gemini-1.5-pro"]
-        self.models = ["gemini-1.0-pro", "gemini-1.5-flash", "gemini-1.5-pro", "01-ai/Yi-1.5-34B-Chat", "google/gemma-1.1-2b-it",\
-                          "google/gemma-1.1-7b-it", 'gemma-7b-it', 'llama3-70b-8192', 'llama3-8b-8192', 'mixtral-8x7b-32768']
+        self.models = [
+    "gemini-pro", "gemini-pro-vision", 
+    "01-ai/Yi-1.5-34B-Chat", "google/gemma-1.1-2b-it", 
+    "gemma-7b-it", "llama3-70b-8192", "llama3-8b-8192", "mixtral-8x7b-32768"]    
+
 
     def call_groq(self, model, temp = 0.7, given_prompt = "Hi"):
         try: