Update app.py

app.py

@@ -11,7 +11,15 @@ class Task(BaseModel):
 
 def make_llm():
     # LangChain helper; will raise if GOOGLE_API_KEY missing
-    return ChatGoogleGenerativeAI(model="gemini-2.0-flash-exp")
+    # return ChatGoogleGenerativeAI(model="gemini-2.0-flash-exp")
+    
+    # Allow override from HF Secrets, but fall back to a stable, supported model
+    model_id = os.getenv("GEMINI_MODEL", "gemini-2.0-flash")
+    return ChatGoogleGenerativeAI(
+        model=model_id,
+        max_retries=3,          # built-in exponential back-off
+        temperature=0.2,
+    )
 
 @app.post("/run")
 async def run_task(t: Task):