Tsunnami commited on
Commit
64c5436
1 Parent(s): 2bceb6f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -4
README.md CHANGED
@@ -443,8 +443,8 @@ topic_model = BERTopic(
443
  language="english",
444
  embedding_model=embedding_model,
445
  umap_model=umap_model,
446
- #hdbscan_model=hdbscan_model,
447
- #vectorizer_model=vectorizer_model,
448
  representation_model=representation_models,
449
  verbose=True,
450
  )
@@ -529,7 +529,6 @@ class SummarizationRepresentation(BaseRepresentation):
529
  ) -> Mapping[str, List[Tuple[str, float]]]:
530
  updated_topics = {}
531
  for topic_id, words in topics.items():
532
- # Extract only the words from the tuples
533
  words_only = [word[0] for word in words]
534
  text = " ".join(words_only)
535
  summary = summarize_with_model(text)
@@ -537,4 +536,4 @@ class SummarizationRepresentation(BaseRepresentation):
537
  return updated_topics
538
 
539
  summarization = SummarizationRepresentation(summarization_model, summarization_tokenizer)
540
- ```
 
443
  language="english",
444
  embedding_model=embedding_model,
445
  umap_model=umap_model,
446
+ hdbscan_model=hdbscan_model,
447
+ vectorizer_model=vectorizer_model,
448
  representation_model=representation_models,
449
  verbose=True,
450
  )
 
529
  ) -> Mapping[str, List[Tuple[str, float]]]:
530
  updated_topics = {}
531
  for topic_id, words in topics.items():
 
532
  words_only = [word[0] for word in words]
533
  text = " ".join(words_only)
534
  summary = summarize_with_model(text)
 
536
  return updated_topics
537
 
538
  summarization = SummarizationRepresentation(summarization_model, summarization_tokenizer)
539
+ ```