Update app.py

app.py

@@ -252,8 +252,12 @@ async def process_document_pipeline(
         
         # Step 1: NER Processing
         logger.info(f"🔍 Running NER for session {session_id}")
-        ner_results = extract_legal_entities(text)
-        
+        ner_results = extract_legal_entities(
+            text,
+            model_id="kn29/my-ner-model",
+            hf_token=os.getenv("HF_TOKEN")  # optional, if your model is private
+        )
+                
         # Store NER results
         await db.ner_results.insert_one({
             "session_id": session_id,