Update app.py

app.py

@@ -270,7 +270,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from PyPDF2 import PdfReader
 from peft import get_peft_model, LoraConfig, TaskType
 
-# ✅ Force CPU execution for Hugging Face Spaces
+# ✅ Force CPU execution
 device = torch.device("cpu")
 
 # 🔹 Load IBM Granite Model (CPU-Compatible)
@@ -370,7 +370,9 @@ def main():
     # 🔹 File Upload Section
     uploaded_file = st.file_uploader("📂 Upload a contract document (PDF)", type="pdf")
 
-    if uploaded_file:
+    # ✅ Ensure file upload message is displayed
+    if uploaded_file is not None:
+        st.session_state["uploaded_file"] = uploaded_file  # Persist file in session state
         st.success("✅ File uploaded successfully!")
         st.write("Click the button below to analyze the contract.")
 
@@ -393,3 +395,4 @@ if __name__ == '__main__':
     main()
 
 
+