as-cle-bert's picture
Update app.py
8abdea8 verified
raw
history blame
563 Bytes
import gradio as gr
import spaces
from load_model import pipe
from utils import Translation
@spaces.GPU
def reply(message, history):
txt = Translation(message, "en")
if txt.original == "en":
response = pipe(message)
return response[0]["generated_text"]
else:
translation = txt.translatef()
response = pipe(translation)
t = Translation(response[0]["generated_text"], txt.original)
res = t.translatef()
return res
demo = gr.ChatInterface(fn=reply, title="Multilingual-Bloom Bot")
demo.launch()