Spaces:
Sleeping
Sleeping
Gaurav vashistha
commited on
Commit
Β·
50e246b
1
Parent(s):
1900bac
Prioritize gemini-1.5-flash-001 to avoid 404s
Browse files- agents/visual_analyst.py +9 -8
agents/visual_analyst.py
CHANGED
|
@@ -17,19 +17,20 @@ class VisualAnalyst:
|
|
| 17 |
# --- SMART MODEL SELECTION ---
|
| 18 |
print("π Checking available Gemini models for this key...")
|
| 19 |
try:
|
|
|
|
| 20 |
my_models = [m.name for m in genai.list_models() if 'generateContent' in m.supported_generation_methods]
|
| 21 |
print(f"π Available Models: {my_models}")
|
| 22 |
|
| 23 |
-
#
|
| 24 |
preferred_order = [
|
| 25 |
-
'models/gemini-1.5-flash',
|
| 26 |
-
'models/gemini-1.5-flash
|
| 27 |
-
'models/gemini-1.5-pro',
|
| 28 |
'models/gemini-1.5-pro-001',
|
| 29 |
-
'models/gemini-pro
|
|
|
|
| 30 |
]
|
| 31 |
|
| 32 |
-
selected_model = "models/gemini-1.5-flash" # Default
|
| 33 |
|
| 34 |
for model_name in preferred_order:
|
| 35 |
if model_name in my_models:
|
|
@@ -40,8 +41,8 @@ class VisualAnalyst:
|
|
| 40 |
self.model = genai.GenerativeModel(selected_model)
|
| 41 |
|
| 42 |
except Exception as e:
|
| 43 |
-
print(f"β οΈ Model list failed ({e}), defaulting to gemini-1.5-flash")
|
| 44 |
-
self.model = genai.GenerativeModel('gemini-1.5-flash')
|
| 45 |
|
| 46 |
async def analyze_image(self, image_path: str):
|
| 47 |
# Adaptation: Read file path to bytes, as main.py passes a path
|
|
|
|
| 17 |
# --- SMART MODEL SELECTION ---
|
| 18 |
print("π Checking available Gemini models for this key...")
|
| 19 |
try:
|
| 20 |
+
# Filter for models that support content generation
|
| 21 |
my_models = [m.name for m in genai.list_models() if 'generateContent' in m.supported_generation_methods]
|
| 22 |
print(f"π Available Models: {my_models}")
|
| 23 |
|
| 24 |
+
# UPDATED PRIORITY: Specific versions FIRST
|
| 25 |
preferred_order = [
|
| 26 |
+
'models/gemini-1.5-flash-001', # <--- Specific version (Safest)
|
| 27 |
+
'models/gemini-1.5-flash', # Alias (Fallback)
|
|
|
|
| 28 |
'models/gemini-1.5-pro-001',
|
| 29 |
+
'models/gemini-1.5-pro',
|
| 30 |
+
'models/gemini-pro-vision'
|
| 31 |
]
|
| 32 |
|
| 33 |
+
selected_model = "models/gemini-1.5-flash-001" # Default safety
|
| 34 |
|
| 35 |
for model_name in preferred_order:
|
| 36 |
if model_name in my_models:
|
|
|
|
| 41 |
self.model = genai.GenerativeModel(selected_model)
|
| 42 |
|
| 43 |
except Exception as e:
|
| 44 |
+
print(f"β οΈ Model list failed ({e}), defaulting to gemini-1.5-flash-001")
|
| 45 |
+
self.model = genai.GenerativeModel('gemini-1.5-flash-001')
|
| 46 |
|
| 47 |
async def analyze_image(self, image_path: str):
|
| 48 |
# Adaptation: Read file path to bytes, as main.py passes a path
|