Th3BossC commited on
Commit
36c2546
·
1 Parent(s): 34f203c

tried something

Browse files
Files changed (1) hide show
  1. qnabackend/common/utils.py +3 -3
qnabackend/common/utils.py CHANGED
@@ -21,10 +21,10 @@ def getText(url : str):
21
  return scraped_text
22
 
23
 
24
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
25
 
26
- model = AutoModelForSeq2SeqLM.from_pretrained('google/flan-t5-large')
27
- tokenizer = AutoTokenizer.from_pretrained('google/flan-t5-large')
28
 
29
  def getAnswer(url : str, question : str):
30
  context = getText(url)
 
21
  return scraped_text
22
 
23
 
24
+ from transformers import T5Tokenizer, T5ForConditionalGeneration
25
 
26
+ model = T5ForConditionalGeneration.from_pretrained('google/flan-t5-large')
27
+ tokenizer = T5Tokenizer.from_pretrained('google/flan-t5-large')
28
 
29
  def getAnswer(url : str, question : str):
30
  context = getText(url)