Spaces:

TarekAi
/

Brainstorming

Runtime error

taitbaha commited on Jun 6

Commit

20f7fc9

•

1 Parent(s): 4e5a700

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from bertopic import BERTopic
-topic_model = BERTopic.load("TarekAi/ArBERTopic")
 def predict_topic(document):
     topics, probabilities = topic_model.transform([document])

 import gradio as gr
 from bertopic import BERTopic
+topic_model = BERTopic.load("davanstrien/label_model")
 def predict_topic(document):
     topics, probabilities = topic_model.transform([document])