Upload folder using huggingface_hub

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 
-model_name = "dbmdz/distilbert-base-cased-finetuned-conll03-english"
+model_name = "dslim/bert-base-NER"
 
 def ner_analysis(text):
     ner_pipeline = pipeline('ner', model=model_name, use_auth_token=None, aggregation_strategy="max")
@@ -11,7 +11,7 @@ def ner_analysis(text):
 
 with gr.Blocks() as demo:
     gr.Markdown("# Named Entity Recognition")
-    gr.Markdown("Powered by dbmdz/distilbert-base-cased-finetuned-conll03-english")
+    gr.Markdown("Powered by dslim/bert-base-NER")
     text_input = gr.Textbox(label="Text", placeholder="Enter text to analyze entities...", lines=3)
     output = gr.Textbox(label="Entities", lines=5)
     btn = gr.Button("Analyze")