Changed zero shot min similarity to 0.5

app.py

@@ -185,7 +185,7 @@ def extract_topics(in_files, in_file, min_docs_slider, in_colnames, max_topics_s
                                 min_topic_size = min_docs_slider,
                                 nr_topics = max_topics_slider,
                                 zeroshot_topic_list = zero_shot_topics_lower,
-                                zeroshot_min_similarity = 0.7,
+                                zeroshot_min_similarity = 0.5,#0.7,
                                 representation_model=representation_model,
                                 verbose = True)
         

funcs/representation_model.py

@@ -118,6 +118,8 @@ llm_config = LLamacppInitConfigGpu(last_n_tokens_size=last_n_tokens_size,
 ## Create representation model parameters ##
 # KeyBERT
 keybert = KeyBERTInspired(random_state=random_seed)
+# MMR
+mmr = MaximalMarginalRelevance(diversity=0.3)
 
 def create_representation_model(create_llm_topic_labels, llm_config, hf_model_name, hf_model_file, chosen_start_tag):
 
@@ -141,6 +143,7 @@ def create_representation_model(create_llm_topic_labels, llm_config, hf_model_na
 
     elif create_llm_topic_labels == "No":
         representation_model = {"KeyBERT": keybert}
+        #representation_model = {"mmr": mmr}
 
     # Deprecated example using CTransformers. This package is not really used anymore
     #model = AutoModelForCausalLM.from_pretrained('NousResearch/Nous-Capybara-7B-V1.9-GGUF', model_type='mistral', model_file='Capybara-7B-V1.9-Q5_K_M.gguf', hf=True, **vars(llm_config))