Update app.py

app.py

@@ -186,7 +186,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             while (answer is None) and (tries < 100): 
                 print(f"try {tries}")
                 maybe_answer = agent(question_text, file_name)
-                if isinstance(maybe_answer, str) and maybe_answer.startswith("Error in generating final LLM output: 504 Server Error: Gateway Time-out for"):
+                if isinstance(maybe_answer, str) and maybe_answer.startswith("Error in generating final LLM output"):
                     tries += 1
                     time.sleep(10)
                 else: