hf-dongpyo commited on
Commit
7f79884
1 Parent(s): 4c740cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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('KETI-AIR/ke-t5-small')
16
- mdl = T5ForConditionalGeneration.from_pretrained('KETI-AIR/ke-t5-small')
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)