models / app.py
Nadasr's picture
Create app.py
e39313b verified
import gradio as gr
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
model_id = "Nadasr/sentAnalysisModel"
classifier = pipeline(
"text-classification",
model=model_id,
tokenizer=model_id,
return_all_scores=False
)
label_map = {
"LABEL_0": "سلبي",
"LABEL_1": "إيجابي",
"0": "سلبي",
"1": "إيجابي",
}
def predict(text):
text = text.strip()
if not text:
return "اكتب الجملة أولاً 🙂"
result = classifier(text)[0]
label = label_map.get(result["label"], result["label"])
score = round(float(result["score"]), 3)
return f"{label} (score = {score})"
demo = gr.Interface(
fn=predict,
inputs=gr.Textbox(lines=3, label="النص العربي"),
outputs=gr.Textbox(label="نتيجة التحليل"),
title="نموذج تحليل المشاعر بالعربية",
description="أدخل الجملة وسيتم تصنيفها إلى إيجابي أو سلبي."
)
if __name__ == "__main__":
demo.launch()