Update app.py

app.py

@@ -47,7 +47,7 @@ def translate_text(text, target_language):
             return text
     return text  # If English, return the same text
 
-# Function to send a request to the Hugging Face API with retry on 503
+# Function to send a request to the Hugging Face API with retry on 503 and error handling for 500
 def generate_speech(model_id, text):
     url = f"https://api-inference.huggingface.co/models/{model_id}"
     headers = {"Authorization": f"Bearer {HUGGINGFACE_API_TOKEN}"}
@@ -64,6 +64,9 @@ def generate_speech(model_id, text):
             estimated_time = response_data.get("estimated_time", 10)
             st.info(f"Model is loading, please wait {estimated_time} seconds...")
             time.sleep(estimated_time)  # Wait before retrying
+        elif response.status_code == 500:
+            st.error("Internal Server Error: The model encountered an issue. Please try again later.")
+            return None
         else:
             st.error(f"Error: {response.status_code} - {response.text}")
             return None
@@ -101,7 +104,7 @@ if submit and user_input:
             st.write(f"Original Text: {user_input}")
             st.write(f"Translated Text: {translated_text}")
 
-        # Generate speech with retry if the model is loading
+        # Generate speech with retry if the model is loading and handle 500 errors
         audio_data = generate_speech("myshell-ai/MeloTTS-English", translated_text)
 
         if audio_data: