Rupesx007 commited on
Commit
b69329e
·
verified ·
1 Parent(s): daeedb3

updated app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ class QuizGenerator:
27
  self.summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
28
 
29
  # Load question generation model
30
- self.qg_model = AutoModelForSeq2SeqGeneration.from_pretrained("valhalla/t5-base-qg-hl")
31
  self.qg_tokenizer = AutoTokenizer.from_pretrained("valhalla/t5-base-qg-hl")
32
 
33
  # Initialize MCQ generation components
 
27
  self.summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
28
 
29
  # Load question generation model
30
+ self.qg_model = AutoModelForSeq2SeqLM.from_pretrained("valhalla/t5-base-qg-hl")
31
  self.qg_tokenizer = AutoTokenizer.from_pretrained("valhalla/t5-base-qg-hl")
32
 
33
  # Initialize MCQ generation components