changing to use dsfsi

app.py

@@ -16,10 +16,8 @@ Birinci Dünya Savaşı sırasında Osmanlı ordusunda görev yapan Atatürk, Ç
 
 st.title("Demo for Sestwana NER Models")
 
-model_list = ['https://huggingface.co/dsfsi/PuoBERTa',
-             'akdeniz27/convbert-base-turkish-cased-ner',
-             'xlm-roberta-large-finetuned-conll03-english',
-             'asahi417/tner-xlm-roberta-base-ontonotes5']
+model_list = ['https://huggingface.co/dsfsi/PuoBERTa'
+             'akdeniz27/convbert-base-turkish-cased-ner',]
 
 st.sidebar.header("Select NER Model")
 model_checkpoint = st.sidebar.radio("", model_list)
@@ -50,7 +48,7 @@ elif input_method == "Write or Paste New Text":
 def setModel(model_checkpoint, aggregation):
     tokenizer = AutoTokenizer.from_pretrained("dsfsi/PuoBERTa-NER")
     model = AutoModelForTokenClassification.from_pretrained("dsfsi/PuoBERTa-NER")
-    return pipeline('ner', model=model, tokenizer=tokenizer, aggregation_strategy=aggregation)
+    return pipeline("token-classification", model="dsfsi/PuoBERTa-NER")
 
 @st.cache_resource
 def get_html(html: str):