Broomva's picture
fixed return value
6e78182
raw
history blame contribute delete
445 Bytes
import random
import time
import gradio as gr
from transformers import pipeline
translator = pipeline("translation", model=f'Broomva/bart-large-translation-spa-guc')
def slow_echo(message, history):
translated = translator(message)[0]['translation_text']
for i in range(len(translated)):
time.sleep(0.05)
yield translated[: i+1]
demo = gr.ChatInterface(slow_echo).queue()
if __name__ == "__main__":
demo.launch()