Update app.py

app.py

@@ -1,27 +1,14 @@
+from gradio_client import Client
 import gradio as gr
-import os
-import requests
-import time
 
-
-API_URL = "https://api-inference.huggingface.co/models/atlasia/Terjman-Ultra"
-HF_TOKEN = os.environ['HF_TOKEN']
-headers = {"Authorization": "Bearer "+ HF_TOKEN}
+client = Client("https://4f122e3ce62b0d809b.gradio.live/")
 
 def respond(english_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)
+    darija_translated_text = client.predict(
+    		text=english_text,
+    		api_name="/predict"
+    )
+    return darija_translated_text
 
 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) 🤗")