Update app.py

app.py

@@ -16,9 +16,13 @@ st.sidebar.image(image, caption=None, width=None, use_column_width=None, clamp=F
 @st.cache_resource
 def load_model(version):
     if version == "version base":
-        return 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-NER-base-frenchNER', tokenizer='bourdoiscatie/Camembert-NER-base-frenchNER', grouped_entities=True)
+        result = ner(text)
+        return result   
     else:
-        return 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-NER-base-frenchNER', tokenizer='bourdoiscatie/Camembert-NER-base-frenchNER', grouped_entities=True)
+        result = ner(text)
+        return result   
 
 def getcolor(texts, labels):
     colors = {'LOC': '#8ef', 'PER': '#faa', 'ORG': '#afa'}