Shriharsh commited on
Commit
0085c9f
·
verified ·
1 Parent(s): 28bf895

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -6
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
- preferred_model = "gemini-1.5-flash-latest"
 
99
  if preferred_model in available_models:
100
  gemini_model = genai.GenerativeModel(preferred_model)
101
  print(f"Using model: {preferred_model}")
102
  else:
103
- fallback_model = "gemini-1.5-pro-latest"
104
- if fallback_model in available_models:
105
- gemini_model = genai.GenerativeModel(fallback_model)
106
- print(f"Using model: {fallback_model}")
 
 
107
  else:
108
- raise ValueError("No suitable free-tier Gemini model available. Available models: " + str(available_models))
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):