ganeshkamath89 commited on
Commit
7ee34df
·
verified ·
1 Parent(s): 84cbb2e

Copying code from VS Code

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -2,9 +2,8 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  def generate_story(story):
5
- ner = pipeline("ner", model = "dbmdz/bert-large-cased-finetuned-conll03-english")
6
- entities = ner(story)
7
- return entities[0]['entity']
8
 
9
  demo = gr.Interface (
10
  fn=generate_story,
 
2
  from transformers import pipeline
3
 
4
  def generate_story(story):
5
+ ner = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english")
6
+ return str(ner(story))
 
7
 
8
  demo = gr.Interface (
9
  fn=generate_story,