|
import gradio as gr |
|
import requests |
|
import os |
|
|
|
API_URL = "https://api-inference.huggingface.co/models/atlasia/Terjman-Large-v2" |
|
HF_TOKEN = os.environ['HF_TOKEN'] |
|
headers = {"Authorization": "Bearer "+ HF_TOKEN} |
|
|
|
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) |
|
else: |
|
return "An error occurred, please refresh the webpage: " + str(response) |
|
|
|
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() |
