Spaces:
Sleeping
Sleeping
import gradio as gr | |
from bertopic import BERTopic | |
model = BERTopic.load("AlexanderHolmes0/C1-Topic-Model-100") | |
model2 = BERTopic.load("AlexanderHolmes0/C2-Topic-Model-100") | |
with gr.Blocks() as demo: | |
with gr.Tab("C1 Topic Model"): | |
gr.load("models/AlexanderHolmes0/C1-Topic-Model-100") | |
with gr.Row(): | |
gr.Plot(model.visualize_topics(width=650),min_width=650) | |
gr.Plot(model.visualize_hierarchy(width=650,height=650),min_width=650) | |
with gr.Tab("C2 Topic Model"): | |
gr.load("models/AlexanderHolmes0/C2-Topic-Model-100",alias = "C2 Topic Model") | |
with gr.Row(): | |
gr.Plot(model2.visualize_topics(width=650),min_width=650) | |
gr.Plot(model2.visualize_hierarchy(width=650,height=650),min_width=650) | |
demo.launch() |