update app

app.py

@@ -109,7 +109,7 @@ st.markdown(
     unsafe_allow_html=True,
 )
 
-# Initialization for Legal NER
+# Initialization for German Legal NER
 tokenizer_legal = AutoTokenizer.from_pretrained("PaDaS-Lab/gbert-legal-ner")
 model_legal = AutoModelForTokenClassification.from_pretrained(
     "PaDaS-Lab/gbert-legal-ner"
@@ -123,7 +123,7 @@ model_gdpr = AutoModelForTokenClassification.from_pretrained(
 )
 ner_gdpr = pipeline("ner", model=model_gdpr, tokenizer=tokenizer_gdpr)
 
-# Define class labels for Legal and GDPR NER models
+# Define class labels for German Legal and GDPR NER models
 classes_legal = {
     "AN": "Lawyer",
     "EUN": "European legal norm",
@@ -220,11 +220,11 @@ st.markdown("<hr>", unsafe_allow_html=True)
 
 test_sentence = st.text_area("Enter Text:", height=150)
 model_choice = st.selectbox(
-    "Choose a model:", ["Legal NER", "GDPR Privacy Policy NER"], index=0
+    "Choose a model:", ["German Legal NER", "GDPR Privacy Policy NER"], index=0
 )
 
 if st.button("Analyze"):
-    if model_choice == "Legal NER":
+    if model_choice == "German Legal NER":
         ner_model = ner_legal
         current_classes = classes_legal
         current_ner_labels = ner_labels_legal