GiladtheFixer commited on
Commit
8f6a211
·
1 Parent(s): fecc2d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
 
5
 
6
  def predict(text,name):
7
- pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-es")
8
  return pipe(text)[0]["translation_text"]
9
 
10
  demo = gr.Interface(
 
4
 
5
 
6
  def predict(text,name):
7
+ pipe = pipeline("translation", model=name)
8
  return pipe(text)[0]["translation_text"]
9
 
10
  demo = gr.Interface(