Update app.py

app.py

@@ -1,15 +1,24 @@
 from gradio_client import Client
 import gradio as gr
 
-client = Client("https://83db3eb39c953bd85d.gradio.live/")
+API_URL = "https://api-inference.huggingface.co/models/atlasia/Terjman-Ultra"
+HF_TOKEN = os.environ['HF_TOKEN']
+headers = {"Authorization": "Bearer "+ HF_TOKEN}
 
 def respond(english_text):
-    darija_translated_text = client.predict(
-    		text=english_text,
-    		api_name="/predict"
-    )
-    return darija_translated_text
-
+    payload = {
+        "inputs": english_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-Ultra 👨‍💻🤩", description="Translate English text to Moroccan Darija using our second biggest model (1.3B) 🤗")