Spaces:

Pranjal12345
/

QAGenerator_with_text

Sleeping

Pranjal12345 commited on Nov 10, 2023

Commit

0be2832

•

1 Parent(s): f060475

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import re
 import os
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-tokenizer = AutoTokenizer.from_pretrained("potsawee/t5-large-generation-race-QuestionAnswer")
-model = AutoModelForSeq2SeqLM.from_pretrained("potsawee/t5-large-generation-race-QuestionAnswer")
 def generate_question_answer_pairs(input_text):
     if input_text is None:

 import os
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = AutoTokenizer.from_pretrained("potsawee/t5-large-generation-squad-QuestionAnswer")
+model = AutoModelForSeq2SeqLM.from_pretrained("potsawee/t5-large-generation-squad-QuestionAnswer")
 def generate_question_answer_pairs(input_text):
     if input_text is None: