Spaces:
Runtime error
Runtime error
Try to print entity from 0th element
Browse files
app.py
CHANGED
@@ -4,8 +4,7 @@ from transformers import pipeline
|
|
4 |
def generate_story(story):
|
5 |
ner = pipeline("ner", model = "dbmdz/bert-large-cased-finetuned-conll03-english")
|
6 |
entities = ner(story)
|
7 |
-
|
8 |
-
return story;
|
9 |
|
10 |
demo = gr.Interface (
|
11 |
fn=generate_story,
|
|
|
4 |
def generate_story(story):
|
5 |
ner = pipeline("ner", model = "dbmdz/bert-large-cased-finetuned-conll03-english")
|
6 |
entities = ner(story)
|
7 |
+
return str(entities[0]['entity'])
|
|
|
8 |
|
9 |
demo = gr.Interface (
|
10 |
fn=generate_story,
|