Update app.py

app.py

@@ -14,9 +14,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 embedding_model = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
 db = FAISS.load_local("faiss_index_lawcounsel", embedding_model, allow_dangerous_deserialization=True)
 
-# Load CUAD chunks for context (optional if index already covers this)
-with open("CUADv1.json", "r") as f:
-    raw_data = json.load(f)
+# Load CUAD from hf itself 
+from datasets import load_dataset
+cuad_data = load_dataset("cuad")
 
 # Model setup flags
 USE_LLAMA = os.environ.get("USE_LLAMA", "false").lower() == "true"