Update app.py

app.py

@@ -7,13 +7,21 @@ HF_TOKEN = os.environ['HF_TOKEN']
 headers = {"Authorization": "Bearer "+ HF_TOKEN}
 
 def respond(english_text):
-    payload  = {
+    payload = {
         "inputs": english_text,
     }
-    darija_translated_text = requests.post(API_URL, headers=headers, json=payload)
-    return darija_translated_text.json()[0]["generated_text"]
+    
+    while True:
+        response = requests.post(API_URL, headers=headers, json=payload).json()
+        
+        if isinstance(response, list) and 'generated_text' in response[0]:
+            return response[0]['generated_text']
+        elif 'estimated_time' in response:
+            time.sleep(5)  # Wait for 5 seconds before retrying
+        else:
+            return "An error occurred, please refresh the webpage: " + str(response)
 
-app = gr.Interface(fn=respond, inputs="text", outputs="text", title="Terjman-Large 👨‍💻🥰", description="Translate English text to Moroccan Darija using our Large model (240M) 🤗")
+app = gr.Interface(fn=respond, inputs="text", outputs="text", title="Terjman-Large-v2 👨‍💻🥰", description="Translate English text to Moroccan Darija using our Large model (240M) 🤗")
 
 if __name__ == "__main__":
     app.launch()