Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import requests
 import os
+import time  # Import time for sleep
 
 API_URL = "https://api-inference.huggingface.co/models/ehcalabres/wav2vec2-lg-xlsr-en-speech-emotion-recognition"
 headers = {"Authorization": f"Bearer {os.getenv('HF_API_TOKEN')}"}
@@ -17,11 +18,20 @@ emotion_to_neg_rate = {
 }
 
 def get_emotion(audio_file):
-    response = requests.post(API_URL, headers=headers, files={"file": audio_file})
-    if response.status_code == 200:
-        return response.json()
-    else:
-        return {"error": response.text}
+    for _ in range(5):  # Retry up to 5 times
+        response = requests.post(API_URL, headers=headers, files={"file": audio_file})
+        if response.status_code == 200:
+            return response.json()
+        elif response.status_code == 503:  # Service Unavailable
+            error_info = response.json()
+            if "error" in error_info and "loading" in error_info["error"]:
+                st.write("Model is currently loading. Please wait...")
+                time.sleep(10)  # Wait for 10 seconds before retrying
+            else:
+                return {"error": response.text}
+        else:
+            return {"error": response.text}
+    return {"error": "Model is taking too long to load or an unexpected error occurred."}
 
 def get_neg_rate(emotion):
     return emotion_to_neg_rate.get(emotion, 30.0)