Spaces:
Runtime error
Runtime error
removing str function to string
Browse files
app.py
CHANGED
@@ -4,7 +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 |
-
return
|
8 |
|
9 |
demo = gr.Interface (
|
10 |
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 entities[0]['entity']
|
8 |
|
9 |
demo = gr.Interface (
|
10 |
fn=generate_story,
|