Update app.py

app.py

@@ -4,15 +4,21 @@ from bertopic import BERTopic
 
 topic_model = BERTopic.load("TarekAi/ArBERTopic")
 
-topic, prob = topic_model.transform("عدم إنجاز الواجبات المدرسية")
+def predict_topic(document):
+    topics, probabilities = topic_model.transform([document])
+    return {"Predicted Topic": topics[0], "Probabilities": probabilities[0].tolist()}
 
-examples=["عدم الالتزام بالتشريع المنظم للحياة المدرسية من طرف الاستاذ", "عدم إنجاز الواجبات المدرسية"]
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=predict_topic,
+    inputs=gr.inputs.Textbox(lines=5, placeholder="Enter your idea here..."),
+    outputs=[
+        gr.outputs.Textbox(label="Predicted Topic"),
+        gr.outputs.Label(label="Topic Probabilities")
+    ],
+    title="BERTopic Inference",
+    description="Enter a single idea to get the predicted topic and its probabilities."
+)
 
-io = gr.Interface(fn=text_classification, 
-                         inputs= gr.Textbox(lines=2, label="Text", placeholder="Enter your idea here..."), 
-                         outputs=gr.Textbox(lines=2, label="Text Classification Result"),
-                         title="Text Classification",
-                         description="Enter a text and see the text classification result!",
-                         examples=examples)
-
-io.launch()
+# Launch the interface
+iface.launch()