Spaces:

Manu101
/

opus-mt-en-de

Sleeping

Manu101 commited on Jun 8

Commit

036c15a

•

1 Parent(s): adfb0cb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import gradio as gd
 mdl_name = "Helsinki-NLP/opus-mt-en-de"
 tk = AutoTokenizer.from_pretrained(mdl_name)
-model = AutoModelForSeqLM.from_pretrained(mdl_name)
 def translate(text):
     encodings = tk(text, return_tensors="pt")

 mdl_name = "Helsinki-NLP/opus-mt-en-de"
 tk = AutoTokenizer.from_pretrained(mdl_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(mdl_name)
 def translate(text):
     encodings = tk(text, return_tensors="pt")