Update app.py

app.py

@@ -1,13 +1,21 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TextIteratorStreamer
 from huggingface_hub import login
-from peft import PeftModel, PeftConfig
 from threading import Thread
 import PyPDF2
 import pandas as pd
 import torch
 import time
 
+# Check if 'peft' is installed
+try:
+    from peft import PeftModel, PeftConfig
+except ImportError:
+    raise ImportError(
+        "The 'peft' library is required but not installed. "
+        "Please install it using: `pip install peft`"
+    )
+
 # Set page configuration
 st.set_page_config(
     page_title="WizNerd Insp",