Victoria Shevchenko commited on
Commit
7d6269f
·
1 Parent(s): df2c973

added type hints

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.