Upload folder using huggingface_hub

app.py

@@ -7,7 +7,7 @@ models = {
     "sentiment": {"model": "cardiffnlp/twitter-roberta-base-sentiment-latest", "task": "sentiment-analysis"},
     "dialogue": {"model": "microsoft/DialoGPT-medium", "task": "text-generation"},
     "summarizer": {"model": "sshleifer/distilbart-cnn-12-6", "task": "summarization"},
-    "ner": {"model": "dbmdz/distilbert-base-cased-finetuned-conll03-english", "task": "ner"},
+    "ner": {"model": "dslim/bert-base-NER", "task": "ner"},
 }
 
 # Initialize pipelines
@@ -77,7 +77,7 @@ with gr.Blocks() as demo:
         sum_btn.click(summarize_text, [sum_text, sum_max, sum_min], sum_out)
 
     with gr.Tab("Named Entity Recognition"):
-        gr.Markdown("## Named Entity Recognition (dbmdz/distilbert-base-cased-finetuned-conll03-english)")
+        gr.Markdown("## Named Entity Recognition (dslim/bert-base-NER)")
         ner_text = gr.Textbox(lines=3, label="Text")
         ner_btn = gr.Button()
         ner_out = gr.Textbox(lines=5, label="Entities")