AhdCompnay commited on
Commit
581d930
1 Parent(s): 0c969fd

Update topic_modeling.py

Browse files
Files changed (1) hide show
  1. topic_modeling.py +3 -1
topic_modeling.py CHANGED
@@ -9,8 +9,10 @@ from wordcloud_fa import WordCloudFa
9
 
10
  import utils
11
 
 
 
12
  class TopicModeling:
13
- def __init__(self, stopwords_path='./assets/stopwords.txt', specific_stopwords_path='./assets/shahrara_stopwords.txt', embedding_model='paraphrase-multilingual-mpnet-base-v2') -> None:
14
  stopwords = open(stopwords_path).read().splitlines()
15
  specific_stopwords = open(specific_stopwords_path).read().splitlines()
16
  stopwords = stopwords + specific_stopwords
 
9
 
10
  import utils
11
 
12
+ embed_model = os.environ.get("EMBED_MODEL")
13
+
14
  class TopicModeling:
15
+ def __init__(self, stopwords_path='./assets/stopwords.txt', specific_stopwords_path='./assets/shahrara_stopwords.txt', embedding_model= embed_model) -> None:
16
  stopwords = open(stopwords_path).read().splitlines()
17
  specific_stopwords = open(specific_stopwords_path).read().splitlines()
18
  stopwords = stopwords + specific_stopwords