Update app.py

app.py

@@ -49,19 +49,24 @@ def home():
 # ---------------------------------------------------------
 # NER ENDPOINT
 # ---------------------------------------------------------
+# ---------------------------------------------------------
+# NER ENDPOINT (UPDATED)
+# ---------------------------------------------------------
 @app.post("/analyze/ner")
 def analyze_ner(data: TextInput):
     try:
-        results = ner_pipeline(data.text, truncation=True)
-
+        # REMOVED truncation=True to fix the 500 error
+        results = ner_pipeline(data.text, aggregation_strategy="simple")
+        
         persons = []
         locations = []
         organizations = []
 
         for entity in results:
             label = entity["entity_group"]
-            word = entity["word"]
-
+            word = entity["word"].strip()
+            
+            # dslim/bert-base-NER uses these labels:
             if label == "PER":
                 persons.append(word)
             elif label == "LOC":
@@ -74,8 +79,9 @@ def analyze_ner(data: TextInput):
             "locations": list(set(locations)),
             "organizations": list(set(organizations))
         }
-
     except Exception as e:
+        # This will help you see the exact error in HF logs
+        print(f"Internal Error: {str(e)}")
         raise HTTPException(status_code=500, detail=str(e))