Spaces:

Timing0311
/

CloudNativeDDL

Runtime error

Timing0311 commited on Jun 30, 2023

Commit

f2d2809

•

1 Parent(s): afe1ac2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ mdl = MT5ForConditionalGeneration.from_pretrained("google/mt5-small")
 tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
 def translation_CN2EN(text):
-    inp = "translate Chinese to English: " + text
     enc = tokenizer(inp, return_tensors="pt").input_ids
     tokens = mdl.generate(enc)
     response = tokenizer.decode(tokens[0], skip_special_tokens=True)

 tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
 def translation_CN2EN(text):
+    inp = "translate English to German: " + text
     enc = tokenizer(inp, return_tensors="pt").input_ids
     tokens = mdl.generate(enc)
     response = tokenizer.decode(tokens[0], skip_special_tokens=True)