added InternalServerError handling, fixed bug

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from litellm import RateLimitError
+from litellm import RateLimitError, InternalServerError
 
 from time import time, sleep
 from agent import MyAgent
@@ -103,8 +103,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             while cooldown_attempts > 0:
                 try:
                     submitted_answer = agent(question_text, additional_data=additional_data)
+                    cooldown_attempts = 0
                     break
-                except RateLimitError as e:
+                except (RateLimitError, InternalServerError) as e:
                     print(f"cooldown due to {e}")
                     sleep(65)
                     cooldown_attempts -= 1