Update utils/metadata.py

utils/metadata.py

@@ -68,11 +68,11 @@ def extract_metadata(text):
     words = text.split()
     chunks = [" ".join(words[i:i + max_chunk_length]) for i in range(0, len(words), max_chunk_length)]
 
-    ner_metadata = {
+        ner_metadata = {
         "EFFECTIVE_DATE": [],
         "PARTIES": [],
         "GOVERNING_LAW": [],
-        "VENUE": []
+        "JURISDICTION": []
     }
 
     label_mapping = {
@@ -96,6 +96,7 @@ def extract_metadata(text):
     ner_metadata["EFFECTIVE_DATE"] = extract_effective_date(text) or ner_metadata["EFFECTIVE_DATE"]
     ner_metadata["GOVERNING_LAW"] = extract_governing_law(text) or ner_metadata["GOVERNING_LAW"]
     ner_metadata["VENUE"] = extract_venue(text) or ner_metadata["VENUE"]
+    ner_metadata["JURISDICTION"] = extract_venue(text) or ner_metadata["JURISDICTION"]
 
     return ner_metadata