Spaces:
Sleeping
Sleeping
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):
|