MrGanesh commited on
Commit
6029573
·
1 Parent(s): 2a14cb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -16,13 +16,7 @@ def load_model():
16
  return model
17
 
18
  model = load_model()
19
- def predict_fn(text, model):
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), top_n=50)
 
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