14 lines
344 B
Python
14 lines
344 B
Python
from typing import List
|
|
|
|
import spacy
|
|
|
|
|
|
def get_keywords(query: str) -> list[str]:
|
|
# Load the NLP model
|
|
nlp = spacy.load("en_core_web_sm")
|
|
# Process the query
|
|
doc = nlp(query)
|
|
# Extract the nouns and adjectives from the query
|
|
keywords = [token.text for token in doc if token.pos_ in ["NOUN", "ADJ"]]
|
|
return keywords
|