Update main.py

main.py

@@ -194,14 +194,14 @@ def process_with_gemini(model, text):
     }"""
     try:
         
-        resp = client.models.generate_content(model='gemini-2.0-flash-exp', contents=[prompt, text])
+        resp = client.models.generate_content(model='gemini-2.0-flash-thinking-exp', contents=[prompt, text])
         time.sleep(6)  # match your Streamlit rate-limit workaround
         return resp.text
     except Exception as e:
         # retry once on 504
         if hasattr(e, "response") and getattr(e.response, "status_code", None) == 504:
             time.sleep(6)
-            resp = client.models.generate_content(model='gemini-2.0-flash-exp', [prompt, text])
+            resp = client.models.generate_content(model='gemini-2.0-flash-thinking-exp', contents=[prompt, text])
             return resp.text
         raise