| 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() |