Spaces:
Runtime error
Runtime error
Commit
·
4c740cc
1
Parent(s):
7e6d4d1
Update app.py
Browse files
app.py
CHANGED
@@ -44,7 +44,7 @@ def text2text_summary(para):
|
|
44 |
|
45 |
def text2text_translate(text):
|
46 |
inp = "translate English to Korean:: " + text
|
47 |
-
enc = text2text_tkn(
|
48 |
tokens = mdl.generate(**enc)
|
49 |
response = text2text_tkn.batch_decode(tokens)
|
50 |
|
|
|
44 |
|
45 |
def text2text_translate(text):
|
46 |
inp = "translate English to Korean:: " + text
|
47 |
+
enc = text2text_tkn(inp, return_tensors = 'pt')
|
48 |
tokens = mdl.generate(**enc)
|
49 |
response = text2text_tkn.batch_decode(tokens)
|
50 |
|