Update app.py
Browse files
app.py
CHANGED
@@ -95,17 +95,20 @@ available_models = [model.name for model in genai.list_models()]
|
|
95 |
print("Available Gemini models:", available_models)
|
96 |
|
97 |
# Select a free-tier model (prefer gemini-1.5-flash-latest, fallback to gemini-1.5-pro-latest)
|
98 |
-
|
|
|
99 |
if preferred_model in available_models:
|
100 |
gemini_model = genai.GenerativeModel(preferred_model)
|
101 |
print(f"Using model: {preferred_model}")
|
102 |
else:
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
|
|
|
|
107 |
else:
|
108 |
-
raise ValueError("No suitable
|
109 |
|
110 |
# Clean text
|
111 |
def clean_text(text):
|
|
|
95 |
print("Available Gemini models:", available_models)
|
96 |
|
97 |
# Select a free-tier model (prefer gemini-1.5-flash-latest, fallback to gemini-1.5-pro-latest)
|
98 |
+
# Select a free-tier model (prefer gemini-pro, fallback to other available models)
|
99 |
+
preferred_model = "gemini-pro" # Use the generally available model
|
100 |
if preferred_model in available_models:
|
101 |
gemini_model = genai.GenerativeModel(preferred_model)
|
102 |
print(f"Using model: {preferred_model}")
|
103 |
else:
|
104 |
+
# Try other available models (if needed)
|
105 |
+
for model_name in ["gemini-1.5-flash", "gemini-1.5-pro"]:
|
106 |
+
if f"models/{model_name}" in available_models:
|
107 |
+
gemini_model = genai.GenerativeModel(f"models/{model_name}")
|
108 |
+
print(f"Using model: models/{model_name}")
|
109 |
+
break # Use the first available match
|
110 |
else:
|
111 |
+
raise ValueError("No suitable Gemini model available. Available models: " + str(available_models))
|
112 |
|
113 |
# Clean text
|
114 |
def clean_text(text):
|