Update app.py

app.py

@@ -16,15 +16,15 @@ st.sidebar.image(image, caption=None, width=None, use_column_width=None, clamp=F
 
 @st.cache_resource
 def load_model(version,text):
-    if version == "NERmemBERT1-3entities-base:
+    if version == "NERmemBERT1-3entities-base":
         ner = pipeline('token-classification', model='CATIE-AQ/NERmembert-base-3entities', tokenizer='CATIE-AQ/NERmembert-base-3entities', aggregation_strategy="simple")
         result = ner(text)
         return result   
-    if version == "NERmemBERT2-3entities-base:
+    if version == "NERmemBERT2-3entities-base":
         ner = pipeline('token-classification', model='CATIE-AQ/NERmembert2-3entities', tokenizer='CATIE-AQ/NERmembert2-3entities', aggregation_strategy="simple")
         result = ner(text)
         return result   
-    if version == "NERmemBERTa-3entities-base:
+    if version == "NERmemBERTa-3entities-base":
         ner = pipeline('token-classification', model='CATIE-AQ/NERmemberta-3entities', tokenizer='CATIE-AQ/NERmemberta-3entities', aggregation_strategy="simple")
         result = ner(text)
         return result   
@@ -32,15 +32,15 @@ def load_model(version,text):
         ner = pipeline('token-classification', model='CATIE-AQ/NERmembert-large-3entities', tokenizer='CATIE-AQ/NERmembert-large-3entities', aggregation_strategy="simple")
         result = ner(text)
         return result   
-    if version == "NERmemBERT1-4entities-base:
+    if version == "NERmemBERT1-4entities-base":
         ner = pipeline('token-classification', model='CATIE-AQ/NERmembert-base-4entities', tokenizer='CATIE-AQ/NERmembert-base-4entities', aggregation_strategy="simple")
         result = ner(text)
         return result   
-    if version == "NERmemBERT2-4entities-base:
+    if version == "NERmemBERT2-4entities-base":
         ner = pipeline('token-classification', model='CATIE-AQ/NERmembert2-4entities', tokenizer='CATIE-AQ/NERmembert2-4entities', aggregation_strategy="simple")
         result = ner(text)
         return result   
-    if version == "NERmemBERTa-4entities-base:
+    if version == "NERmemBERTa-4entities-base":
         ner = pipeline('token-classification', model='CATIE-AQ/NERmemberta-4entities', tokenizer='CATIE-AQ/NERmemberta-4entities', aggregation_strategy="simple")
         result = ner(text)
         return result