Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ s2v = Sense2Vec().from_disk('s2v_reddit_2015_md/s2v_old')
|
|
14 |
|
15 |
# Function to get important keywords
|
16 |
def get_imp_keywords(content):
|
17 |
-
return [token.text for token in nlp(content) if token.pos_ in [
|
18 |
|
19 |
# Function to generate a question
|
20 |
def get_question(context, answer):
|
|
|
14 |
|
15 |
# Function to get important keywords
|
16 |
def get_imp_keywords(content):
|
17 |
+
return [token.text for token in nlp(content) if token.pos_ in [ "PROPN", "NUM"]]
|
18 |
|
19 |
# Function to generate a question
|
20 |
def get_question(context, answer):
|