wilmerags commited on
Commit
1dcaf3f
·
1 Parent(s): 2d21363

fix: Fix keyword index usage in keyword selection

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -146,8 +146,7 @@ def generate_plot(
146
  cluster_tws = []
147
  for ix, obs in enumerate(encoded_labels):
148
  if obs == label:
149
- cluster_tws.append(tws_cleaned)
150
- st.write(tws_cleaned)
151
  cluster_words = [tw.split(' ') for tw in cluster_tws]
152
  cluster_words = list(set(itertools.chain.from_iterable(cluster_words)))
153
  cluster_embeddings = embed_text(cluster_tws, model)
 
146
  cluster_tws = []
147
  for ix, obs in enumerate(encoded_labels):
148
  if obs == label:
149
+ cluster_tws.append(tws_cleaned[ix])
 
150
  cluster_words = [tw.split(' ') for tw in cluster_tws]
151
  cluster_words = list(set(itertools.chain.from_iterable(cluster_words)))
152
  cluster_embeddings = embed_text(cluster_tws, model)