cstr commited on
Commit
3ac0709
·
verified ·
1 Parent(s): fbaef79

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -217,12 +217,6 @@ VISION_MODELS = {
217
  "GoogleAI": ["gemini-1.5-pro", "gemini-1.0-pro", "gemini-1.5-flash", "gemini-2.0-pro", "gemini-2.5-pro"]
218
  }
219
 
220
- # Add all models with "vl", "vision", "visual" in their name to HF vision models
221
- for model_name in list(HUGGINGFACE_MODELS.keys()):
222
- if any(x in model_name.lower() for x in ["vl", "vision", "visual", "llava"]):
223
- if model_name not in VISION_MODELS["HuggingFace"]:
224
- VISION_MODELS["HuggingFace"].append(model_name)
225
-
226
  # OPENAI MODELS
227
  OPENAI_MODELS = {
228
  "gpt-3.5-turbo": 16385,
@@ -358,6 +352,13 @@ GOOGLEAI_MODELS = {
358
  "gemini-2.5-pro": 2000000,
359
  }
360
 
 
 
 
 
 
 
 
361
  # ==========================================================
362
  # HELPER FUNCTIONS
363
  # ==========================================================
 
217
  "GoogleAI": ["gemini-1.5-pro", "gemini-1.0-pro", "gemini-1.5-flash", "gemini-2.0-pro", "gemini-2.5-pro"]
218
  }
219
 
 
 
 
 
 
 
220
  # OPENAI MODELS
221
  OPENAI_MODELS = {
222
  "gpt-3.5-turbo": 16385,
 
352
  "gemini-2.5-pro": 2000000,
353
  }
354
 
355
+
356
+ # Add all models with "vl", "vision", "visual" in their name to HF vision models
357
+ for model_name in list(HUGGINGFACE_MODELS.keys()):
358
+ if any(x in model_name.lower() for x in ["vl", "vision", "visual", "llava"]):
359
+ if model_name not in VISION_MODELS["HuggingFace"]:
360
+ VISION_MODELS["HuggingFace"].append(model_name)
361
+
362
  # ==========================================================
363
  # HELPER FUNCTIONS
364
  # ==========================================================