File size: 598 Bytes
62b6bd5 2c27dbf 62b6bd5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import gradio as gr
from load_model import pipe
from utils import Translation
def reply(message, history):
txt = Translation(message, "en")
if txt.original == "en":
response = pipe(message)
return response[0]["generated_text"].split("Answer\n")[1]
else:
translation = txt.translatef()
response = pipe(translation)
t = Translation(response[0]["generated_text"].split("Answer\n")[1], txt.original)
res = t.translatef()
return res
demo = gr.ChatInterface(fn=reply, title="Multilingual-Phi Bot")
demo.launch() |