Update app.py

app.py

@@ -17,16 +17,16 @@ st.sidebar.image(image, caption=None, width=None, use_column_width=None, clamp=F
 @st.cache_resource
 def load_model(version,text):
     if version == "version base":
-        ner = pipeline('token-classification', model='bourdoiscatie/Camembert-NER-base-frenchNER', tokenizer='bourdoiscatie/Camembert-NER-base-frenchNER', grouped_entities=True)
+        ner = pipeline('token-classification', model='bourdoiscatie/camembert-base-frenchNER_3entities', tokenizer='bourdoiscatie/camembert-base-frenchNER_3entities', grouped_entities=True)
         result = ner(text)
         return result   
     else:
-        ner = pipeline('token-classification', model='bourdoiscatie/Camembert-NER-base-frenchNER', tokenizer='bourdoiscatie/Camembert-NER-base-frenchNER', grouped_entities=True)
+        ner = pipeline('token-classification', model='bourdoiscatie/camembert-base-frenchNER_4entities', tokenizer='bourdoiscatie/camembert-base-frenchNER_4entities', grouped_entities=True)
         result = ner(text)
         return result   
 
 def getcolor(texts, labels):
-    colors = {'LOC': '#38419D', 'PER': '#BF3131', 'ORG': '#597E52'}
+    colors = {'LOC': '#38419D', 'PER': '#BF3131', 'ORG': '#597E52', 'MISC':'#F1C232'}
     return [(t,l,colors[l]) for t, l in zip(texts, labels)]
 
 def color_annotation(to_print,transcript) :