Removing group entities flag and ranaming to friendlier variable names

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 from transformers import pipeline
 
 def generate_story(story):
-    story_gen = pipeline("ner", "dbmdz/bert-large-cased-finetuned-conll03-english", grouped_entities=True)
-    story_expanded = story_gen(story)
-    return str(story_expanded)
+    ner = pipeline("ner", "dbmdz/bert-large-cased-finetuned-conll03-english")
+    entities = ner(story)
+    return str(entities)
 
 demo = gr.Interface (
     fn=generate_story,