AlexanderHolmes0 commited on
Commit
a1dd328
·
verified ·
1 Parent(s): 3500733

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -1,3 +1,18 @@
1
  import gradio as gr
2
 
3
- gr.load("models/AlexanderHolmes0/C1-Topic-Model-100").launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
 
3
+ model = BERTopic.load("AlexanderHolmes0/C1-Topic-Model-100")
4
+ model2 = BERTopic.load("AlexanderHolmes0/C2-Topic-Model-100")
5
+
6
+ with gr.Blocks() as demo:
7
+ with gr.Tab("C1 Topic Model"):
8
+ gr.load("models/AlexanderHolmes0/C1-Topic-Model-100")
9
+ with gr.Row():
10
+ gr.Plot(model.visualize_topics(width=650),min_width=650)
11
+ gr.Plot(model.visualize_hierarchy(width=650,height=650),min_width=650)
12
+ with gr.Tab("C2 Topic Model"):
13
+ gr.load("models/AlexanderHolmes0/C2-Topic-Model-100",alias = "C2 Topic Model")
14
+ with gr.Row():
15
+ gr.Plot(model2.visualize_topics(width=650),min_width=650)
16
+ gr.Plot(model2.visualize_hierarchy(width=650,height=650),min_width=650)
17
+
18
+ demo.launch()