Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
 import fitz  # PyMuPDF
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
 # Load model directly
 model_name = "openai-community/gpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
 # Function to extract text from PDF
 def extract_text_from_pdf(pdf_file):
@@ -42,7 +42,7 @@ def generate_mcqs(text_chunks, num_questions=5):
             question = question_part.split("Options:")[0].strip()
             options_part = question_part.split("Options:")[1].strip()
             options = options_part.split("\n")
-            
+
             # Ensure four options
             if len(options) < 4:
                 continue