Update app.py

app.py

@@ -7,15 +7,18 @@ from fpdf import FPDF
 import tempfile
 import os
 
-# Load models (you can replace with smaller or local models if needed)
+
+
+st.set_page_config(page_title="AI Study Plan Assistant", layout="wide")
+
+
+# Caching model load
 @st.cache_resource
 def load_models():
     summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
     qa_pipeline = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")
     return summarizer, qa_pipeline
 
-summarizer, qa_pipeline = load_models()
-
 # Text extraction
 def extract_text(uploaded_file):
     ext = os.path.splitext(uploaded_file.name)[-1].lower()