Pranjal12345 commited on
Commit
f060475
1 Parent(s): 4a5a876

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ import re
4
  import os
5
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
6
 
7
- tokenizer = AutoTokenizer.from_pretrained("potsawee/t5-large-generation-squad-QuestionAnswer")
8
- model = AutoModelForSeq2SeqLM.from_pretrained("potsawee/t5-large-generation-squad-QuestionAnswer")
9
 
10
  def generate_question_answer_pairs(input_text):
11
  if input_text is None:
 
4
  import os
5
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
6
 
7
+ tokenizer = AutoTokenizer.from_pretrained("potsawee/t5-large-generation-race-QuestionAnswer")
8
+ model = AutoModelForSeq2SeqLM.from_pretrained("potsawee/t5-large-generation-race-QuestionAnswer")
9
 
10
  def generate_question_answer_pairs(input_text):
11
  if input_text is None: