Update main.py

main.py

@@ -1,13 +1,22 @@
 import chardet
 import spacy
-from gliner_spacy import GLiNERComponent
+
+# ------------------------
+# CONFIGURATION
+# ------------------------
+custom_spacy_config = {
+    "gliner_model": "urchade/gliner_multi_pii-v1",
+    "labels": [
+        "person", "organization", "company", "country",
+        "medical condition", "credit card brand",
+    ],
+    "threshold": 0.39,
+    "style": "ent",
+}
 
 # Load SpaCy and add GLiNER to the pipeline
 nlp = spacy.load("en_core_web_lg")
-nlp.add_pipe("gliner_spacy", config={
-    "labels": ["PERSON", "ORGANIZATION", "LOCATION", "DISEASE"],
-    "model": "urchade/gliner_multi_pii-v1"
-}, last=True)
+nlp.add_pipe("gliner_spacy", config=custom_spacy_config)
 
 def detect_encoding(file_bytes):
     result = chardet.detect(file_bytes)