Spaces:
Sleeping
Sleeping
Victoria Shevchenko commited on
Commit ·
7d6269f
1
Parent(s): df2c973
added type hints
Browse files
app.py
CHANGED
|
@@ -41,7 +41,7 @@ def extract_text_from_pdf(pdf_path: str) -> str:
|
|
| 41 |
return text
|
| 42 |
|
| 43 |
@tool
|
| 44 |
-
def extract_entities(text, desired_entities):
|
| 45 |
"""A tool that extracts entities from a text using spaCy.
|
| 46 |
Args:
|
| 47 |
text: A string representing the text to extract entities from.
|
|
|
|
| 41 |
return text
|
| 42 |
|
| 43 |
@tool
|
| 44 |
+
def extract_entities(text: str, desired_entities: list) -> list:
|
| 45 |
"""A tool that extracts entities from a text using spaCy.
|
| 46 |
Args:
|
| 47 |
text: A string representing the text to extract entities from.
|