ChatGPT updates from logs

app.py

@@ -1,26 +1,27 @@
-from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification
+from transformers import pipeline
 import gradio as gr
 
-# Load the tokenizer and model
+# Load the NER pipeline
 try:
-    tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
-    model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
-    ner_model = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
+    ner_model = pipeline("ner", model="dslim/bert-base-NER", aggregation_strategy="simple")
+    print("Model loaded successfully.")
 except Exception as e:
     ner_model = None
     print(f"Error loading model: {e}")
 
 def extract_named_entities(text):
+    if ner_model is None:
+        return [["Error", "Model not loaded", 0.0]]
+
     if not text.strip():
-        return [{"Entity": "N/A", "Text": "No input provided", "Score": 0.0}]
+        return [["Error", "No input provided", 0.0]]
 
     try:
-        if ner_model is None:
-            raise ValueError("Model failed to load. Check logs for details.")
         entities = ner_model(text)
-        return [{"Entity": ent["entity_group"], "Text": ent["word"], "Score": round(ent["score"], 3)} for ent in entities]
+        # Convert list of dictionaries to list of lists for Gradio compatibility
+        return [[ent["entity_group"], ent["word"], round(ent["score"], 3)] for ent in entities]
     except Exception as e:
-        return [{"Entity": "Error", "Text": str(e), "Score": 0.0}]
+        return [["Error", str(e), 0.0]]
 
 # Define the Gradio interface
 iface = gr.Interface(
@@ -29,11 +30,6 @@ iface = gr.Interface(
     outputs=gr.Dataframe(headers=["Entity", "Text", "Score"], label="Named Entities"),
     title="Named Entity Recognition",
     description="Input some text and get the named entities (like names, locations, organizations).",
-    examples=[
-        ["My name is Emilio."],
-        ["Barack Obama was the President of the United States."],
-        ["OpenAI created ChatGPT."],
-    ]
 )
 
 if __name__ == "__main__":