azunre's picture
Update app.py
e41867f verified
import os
import gradio as gr
from gradio_client import Client
MY_HF_TOKEN_KEY = os.environ["MY_HF_TOKEN_KEY"]
client = Client("Ghana-NLP/test-text-models",hf_token=MY_HF_TOKEN_KEY)
def generate(text, direction):
output = client.predict(text, direction)
return output
title = "GhanaNLP: Test Text Translation Models"
description = """
<b>How to use:</b> Translate Twi, Ga or Ewe text to/from English.
"""
examples = [
["Yaaba jogbaŋŋ mi Misuɔmɔ.","Ga->En"],
["ha mi gbe fioo","Ga->En"],
["kɛ osuɔmɔ aha mi","Ga->En"],
]
gr.Interface(
fn=generate,
inputs=[
gr.Text(label="Input Text"),
gr.Dropdown(
["Twi->En", "En->Twi","En->Ga", "Ga->En","En->Ewe", "Ewe->En",
"Dagbani->En", "En->Dagbani","En->Gurene", "Gurene->En","En->Kusaal", "Kusaal->En"],
label="Direction", info="Select From (Input) -> To (Output) languages"
),
],
outputs=[
gr.Text(label="Output Text"),
],
title=title,
description=description,
examples=examples,
).launch()