update gradio server

Dockerfile

@@ -1,6 +1,6 @@
 # read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
 # you will also find guides on how best to write your Dockerfile
-
+# https://huggingface.co/spaces/sayakpaul/demo-docker-gradio/blob/main/Dockerfile
 FROM python:3.11.5
 
 WORKDIR /code

gradio_ner.py

@@ -1,5 +1,6 @@
-from transformers import pipeline
 import gradio as gr
+import logging
+from transformers import pipeline
 
 examples = [
 """
@@ -76,6 +77,8 @@ model_checkpoint = "aimlnerd/bert-finetuned-legalentity-ner"
 ner_pipeline = pipeline(
     "token-classification", model=model_checkpoint, aggregation_strategy="simple"
 )
+logging.info(f"NER pipeline initialized with checkpoint {model_checkpoint}")
+
 
 def ner(text):
     output = ner_pipeline(text)
@@ -86,5 +89,4 @@ demo = gr.Interface(ner,
              gr.HighlightedText(),
              examples=examples)
 
-demo.launch(share=True)
-print()
+demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -1,6 +1,6 @@
 transformers[torch]==4.36.1
 numpy==1.26.3
-scikit-learn==1.3.2
+#scikit-learn==1.3.2
 matplotlib==3.8.2
 datasets==2.16.1
 evaluate==0.4.1