Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -16,13 +16,7 @@ def load_model():
|
|
16 |
return model
|
17 |
|
18 |
model = load_model()
|
19 |
-
|
20 |
-
kw_extractor = yake.KeywordExtractor(top=50,stopwords=pa_stopwords)
|
21 |
-
candidates = kw_extractor.extract_keywords(text)
|
22 |
-
keyphrases = [candidate[0] for candidate in candidates]
|
23 |
-
#kw_model = KeyBERT(model=kw_model)
|
24 |
-
keywords=model.extract_keywords(text, keyphrases, keyphrase_ngram_range=(1, 3), top_n=50)
|
25 |
-
return keywords
|
26 |
|
27 |
st.title("Patent Text Extractor")
|
28 |
placeholder = st.empty()
|
@@ -58,7 +52,8 @@ kw_extractor = yake.KeywordExtractor(top=50, stopwords=pa_stopwords)
|
|
58 |
candidates = kw_extractor.extract_keywords(text)
|
59 |
candidates = [candidate[0] for candidate in candidates]
|
60 |
from keyphrase_vectorizers import KeyphraseCountVectorizer
|
61 |
-
keywords=model.extract_keywords(text,candidates, keyphrase_ngram_range=(1, 3),
|
|
|
62 |
|
63 |
|
64 |
|
|
|
16 |
return model
|
17 |
|
18 |
model = load_model()
|
19 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
st.title("Patent Text Extractor")
|
22 |
placeholder = st.empty()
|
|
|
52 |
candidates = kw_extractor.extract_keywords(text)
|
53 |
candidates = [candidate[0] for candidate in candidates]
|
54 |
from keyphrase_vectorizers import KeyphraseCountVectorizer
|
55 |
+
keywords=model.extract_keywords(text,candidates, keyphrase_ngram_range=(1, 3),
|
56 |
+
top_n=50,stop_words='english',vectorizer=KeyphraseCountVectorizer())
|
57 |
|
58 |
|
59 |
|