AlexanderHolmes0 commited on
Commit
84d7079
·
verified ·
1 Parent(s): a5c8045

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -9,6 +9,9 @@ with open('C1_msg.pkl', 'rb') as f:
9
  reduced_embeddings = np.load('C1_reduced_embeddings.npy')
10
 
11
  model = BERTopic.load("AlexanderHolmes0/C1-Topic-Model-100")
 
 
 
12
  model2 = BERTopic.load("AlexanderHolmes0/C2-Topic-Model-100")
13
 
14
  with gr.Blocks() as demo:
@@ -21,7 +24,8 @@ with gr.Blocks() as demo:
21
  reduced_embeddings=reduced_embeddings,
22
  hide_annotations = True,
23
  topics = range(0,7),
24
- width = 725),
 
25
  min_width=725)
26
  with gr.Tab("C2 Topic Model"):
27
  gr.load("models/AlexanderHolmes0/C2-Topic-Model-100",alias = "C2 Topic Model")
 
9
  reduced_embeddings = np.load('C1_reduced_embeddings.npy')
10
 
11
  model = BERTopic.load("AlexanderHolmes0/C1-Topic-Model-100")
12
+
13
+ model.set_topic_labels({8: "8_shimmer_religion"})
14
+
15
  model2 = BERTopic.load("AlexanderHolmes0/C2-Topic-Model-100")
16
 
17
  with gr.Blocks() as demo:
 
24
  reduced_embeddings=reduced_embeddings,
25
  hide_annotations = True,
26
  topics = range(0,7),
27
+ width = 725,
28
+ custom_labels=True),
29
  min_width=725)
30
  with gr.Tab("C2 Topic Model"):
31
  gr.load("models/AlexanderHolmes0/C2-Topic-Model-100",alias = "C2 Topic Model")