Update app.py

app.py

@@ -97,8 +97,29 @@ def generate_speech_from_text(message: str, huggingface_token: str) -> None:
     API_URL = "https://api-inference.huggingface.co/models/espnet/kan-bayashi_ljspeech_vits"
     headers = {"Authorization": f"Bearer {huggingface_token}"}
     payloads = {"inputs": message}
-    
+
     response = requests.post(API_URL, headers=headers, json=payloads)
+
+    # Handle model loading
+    if response.status_code == 503:
+        try:
+            info = response.json()
+            wait_time = round(info.get("estimated_time", 10))
+        except Exception:
+            wait_time = 10
+        st.warning(f"Model is loading... please wait ~{wait_time} seconds and try again.")
+        return
+
+    # Handle other errors
+    if response.status_code != 200:
+        try:
+            error_message = response.json().get("error", "Unknown error")
+        except Exception:
+            error_message = "Unknown error"
+        st.error(f"⚠️ Audio generation failed: {response.status_code} - {error_message}")
+        return
+
+    # Save audio if successful
     with open("generated_audio.flac", "wb") as file:
         file.write(response.content)