Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ def ner(text):
|
|
11 |
|
12 |
text=text.lower()
|
13 |
|
14 |
-
prediction
|
15 |
filtered_output = [[{v: k} for d in sublist for k, v in d.items() if v.startswith("B-")] for sublist in prediction]
|
16 |
entities = []
|
17 |
for sublist in filtered_output:
|
|
|
11 |
|
12 |
text=text.lower()
|
13 |
|
14 |
+
prediction= model.predict([text])
|
15 |
filtered_output = [[{v: k} for d in sublist for k, v in d.items() if v.startswith("B-")] for sublist in prediction]
|
16 |
entities = []
|
17 |
for sublist in filtered_output:
|