Update app.py

app.py

@@ -1,6 +1,10 @@
 import streamlit as st
 import fitz  # PyMuPDF
-from transformers import AutoModelForCausalLM, AutoTokenizer
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForQuestionAnswering
+
+tokenizer = AutoTokenizer.from_pretrained("google-bert/bert-large-cased-whole-word-masking-finetuned-squad")
+model = AutoModelForQuestionAnswering.from_pretrained("google-bert/bert-large-cased-whole-word-masking-finetuned-squad")
 
 # Function to extract text from PDF
 def extract_text_from_pdf(pdf_file):
@@ -16,9 +20,6 @@ def generate_mcqs(text, num_questions=5):
     if not text.strip():
         return ["No text extracted from the PDF. Unable to generate MCQs."]
 
-    # Load HuggingFace model
-    tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2-large")
-    model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2-large")
 
     # Tokenize the input text
     inputs = tokenizer(text, return_tensors="pt", max_length=512, truncation=True)