Polo123 commited on
Commit
26ae746
1 Parent(s): 80253e3

Update logic.py

Browse files
Files changed (1) hide show
  1. logic.py +2 -2
logic.py CHANGED
@@ -96,8 +96,8 @@ def get_topics(docs):
96
  """
97
 
98
  representation_model = {
99
- "KeyBERT": KeyBERTInspired(),
100
- "LLM": Llam
101
  }
102
 
103
  # Pre-calculate embeddings
 
96
  """
97
 
98
  representation_model = {
99
+ "KeyBERT": KeyBERTInspired(),
100
+ "LLM": LlamaCPP(llm, prompt=prompt),
101
  }
102
 
103
  # Pre-calculate embeddings