File size: 266 Bytes
ede7c54
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import gradio as gr

from transformers import pipeline


def predict(text,name):
  pipe = pipeline("translation", model=name)
  return pipe(text)[0]["translation_text"]

demo = gr.Interface(
  fn=predict, 
  inputs=['text','text'],
  outputs='text',
)

demo.launch()