Brainstorming / app.py
taitbaha's picture
Create app.py
1fed686 verified
raw
history blame
852 Bytes
import gradio as gr
from transformers import pipeline
from bertopic import BERTopic
topic_model = BERTopic.load("TarekAi/ArBERTopic")
topic, prob = topic_model.transform("عدم إنجاز الواجبات المدرسية")
examples=["عدم الالتزام بالتشريع المنظم للحياة المدرسية من طرف الاستاذ", "عدم إنجاز الواجبات المدرسية"]
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()