Spaces:
Runtime error
Runtime error
hf-dongpyo
commited on
Commit
•
7f79884
1
Parent(s):
4c740cc
Update app.py
Browse files
app.py
CHANGED
@@ -12,8 +12,8 @@ import gradio as grad
|
|
12 |
# mdl = AutoModelWithLMHead.from_pretrained('deep-learning-analytics/wikihow-t5-small')
|
13 |
|
14 |
# translate
|
15 |
-
text2text_tkn = T5Tokenizer.from_pretrained('
|
16 |
-
mdl = T5ForConditionalGeneration.from_pretrained('
|
17 |
|
18 |
def text2text(context, answer):
|
19 |
input_text = "answer: %s context: %s </s>" % (answer, context)
|
|
|
12 |
# mdl = AutoModelWithLMHead.from_pretrained('deep-learning-analytics/wikihow-t5-small')
|
13 |
|
14 |
# translate
|
15 |
+
text2text_tkn = T5Tokenizer.from_pretrained('t5-small')
|
16 |
+
mdl = T5ForConditionalGeneration.from_pretrained('t5-small')
|
17 |
|
18 |
def text2text(context, answer):
|
19 |
input_text = "answer: %s context: %s </s>" % (answer, context)
|