| import gradio as gr |
| from transformers import pipeline |
|
|
| |
| MODEL_ID = "H-Layba/urdu-sentiment-classifier" |
|
|
| classifier = pipeline( |
| "text-classification", |
| model=MODEL_ID |
| ) |
|
|
| |
| EXAMPLES = [ |
| ["یہ فلم بہت اچھی تھی"], |
| ["آج کا دن بہت برا تھا"], |
| ["کھانا لاجواب تھا"], |
| ["خدمت بہت خراب تھی"], |
| ["یہ کتاب بہت مفید ہے"], |
| ["بہت مایوسی ہوئی"], |
| ] |
|
|
| def analyze(text): |
| if not text.strip(): |
| return "", "", 0 |
|
|
| result = classifier(text)[0] |
| label = result["label"] |
| score = result["score"] |
|
|
| emoji = "🟢 مثبت (Positive)" if label == "positive" else "🔴 منفی (Negative)" |
| confidence = round(score * 100, 2) |
| bar = score |
|
|
| return emoji, f"{confidence}%", bar |
|
|
|
|
| |
| css = """ |
| @import url('https://fonts.googleapis.com/css2?family=Noto+Nastaliq+Urdu&family=Inter:wght@400;500;600&display=swap'); |
| |
| body, .gradio-container { |
| background: #0a0a0f !important; |
| font-family: 'Inter', sans-serif; |
| } |
| |
| #title { |
| text-align: center; |
| padding: 2rem 1rem 0.5rem; |
| } |
| |
| #title h1 { |
| font-size: 2rem; |
| font-weight: 600; |
| color: #e4e4f0; |
| letter-spacing: -0.02em; |
| margin-bottom: 0.4rem; |
| } |
| |
| #title p { |
| color: #6b6b8a; |
| font-size: 0.95rem; |
| } |
| |
| .urdu-input textarea { |
| font-family: 'Noto Nastaliq Urdu', serif !important; |
| font-size: 1.3rem !important; |
| direction: rtl !important; |
| text-align: right !important; |
| line-height: 2.2 !important; |
| background: #12121e !important; |
| border: 1px solid #2a2a3e !important; |
| color: #e4e4f0 !important; |
| border-radius: 10px !important; |
| padding: 1rem !important; |
| } |
| |
| .urdu-input textarea:focus { |
| border-color: #7c6dfa !important; |
| box-shadow: 0 0 0 2px rgba(124, 109, 250, 0.15) !important; |
| } |
| |
| .result-label { |
| font-size: 1.4rem !important; |
| font-weight: 600 !important; |
| text-align: center !important; |
| padding: 1rem !important; |
| background: #12121e !important; |
| border: 1px solid #2a2a3e !important; |
| border-radius: 10px !important; |
| color: #e4e4f0 !important; |
| } |
| |
| .confidence-text { |
| font-size: 1.1rem !important; |
| text-align: center !important; |
| color: #7c6dfa !important; |
| font-weight: 500 !important; |
| background: #12121e !important; |
| border: 1px solid #2a2a3e !important; |
| border-radius: 10px !important; |
| } |
| |
| button.primary { |
| background: #7c6dfa !important; |
| border: none !important; |
| color: white !important; |
| font-weight: 500 !important; |
| border-radius: 8px !important; |
| padding: 0.6rem 1.5rem !important; |
| } |
| |
| button.primary:hover { |
| background: #6b5ce7 !important; |
| } |
| |
| .example-btn { |
| font-family: 'Noto Nastaliq Urdu', serif !important; |
| direction: rtl !important; |
| font-size: 1rem !important; |
| } |
| |
| #stats { |
| display: flex; |
| justify-content: center; |
| gap: 2rem; |
| padding: 1rem; |
| margin-top: 1rem; |
| } |
| |
| .stat-card { |
| background: #12121e; |
| border: 1px solid #2a2a3e; |
| border-radius: 10px; |
| padding: 0.8rem 1.5rem; |
| text-align: center; |
| } |
| |
| .stat-num { |
| font-size: 1.4rem; |
| font-weight: 600; |
| color: #7c6dfa; |
| font-family: 'JetBrains Mono', monospace; |
| } |
| |
| .stat-label { |
| font-size: 0.75rem; |
| color: #6b6b8a; |
| margin-top: 2px; |
| } |
| |
| footer { display: none !important; } |
| """ |
|
|
| with gr.Blocks(css=css, theme=gr.themes.Base()) as demo: |
|
|
| gr.HTML(""" |
| <div id="title"> |
| <h1>🇵🇰 Urdu Sentiment Analyser</h1> |
| <p>اردو متن کا جذباتی تجزیہ — Fine-tuned mBERT on 50,000 Urdu reviews</p> |
| </div> |
| <div id="stats"> |
| <div class="stat-card"> |
| <div class="stat-num">81%</div> |
| <div class="stat-label">Accuracy</div> |
| </div> |
| <div class="stat-card"> |
| <div class="stat-num">0.81</div> |
| <div class="stat-label">F1 Score</div> |
| </div> |
| <div class="stat-card"> |
| <div class="stat-num">50K</div> |
| <div class="stat-label">Training rows</div> |
| </div> |
| <div class="stat-card"> |
| <div class="stat-num">mBERT</div> |
| <div class="stat-label">Base model</div> |
| </div> |
| </div> |
| """) |
|
|
| with gr.Row(): |
| with gr.Column(scale=3): |
| text_input = gr.Textbox( |
| label="اردو متن یہاں لکھیں (Enter Urdu text)", |
| placeholder="یہاں اردو میں لکھیں...", |
| lines=4, |
| elem_classes="urdu-input" |
| ) |
| analyze_btn = gr.Button("تجزیہ کریں — Analyze", variant="primary") |
|
|
| with gr.Column(scale=2): |
| label_out = gr.Textbox( |
| label="نتیجہ — Result", |
| interactive=False, |
| elem_classes="result-label" |
| ) |
| confidence_out = gr.Textbox( |
| label="اعتماد — Confidence", |
| interactive=False, |
| elem_classes="confidence-text" |
| ) |
| bar_out = gr.Slider( |
| label="Confidence Score", |
| minimum=0, |
| maximum=1, |
| interactive=False |
| ) |
|
|
| gr.Examples( |
| examples=EXAMPLES, |
| inputs=text_input, |
| label="مثالیں — Try these examples", |
| elem_id="examples" |
| ) |
|
|
| gr.HTML(""" |
| <div style="text-align:center; padding: 1.5rem; color: #3a3a5a; font-size: 0.8rem;"> |
| Built by <a href="https://huggingface.co/H-Layba" style="color:#7c6dfa">H-Layba</a> · |
| Model: <a href="https://huggingface.co/H-Layba/urdu-sentiment-classifier" style="color:#7c6dfa">urdu-sentiment-classifier</a> · |
| Dataset: IMDB Urdu Reviews (50K) |
| </div> |
| """) |
|
|
| analyze_btn.click( |
| fn=analyze, |
| inputs=text_input, |
| outputs=[label_out, confidence_out, bar_out] |
| ) |
| text_input.submit( |
| fn=analyze, |
| inputs=text_input, |
| outputs=[label_out, confidence_out, bar_out] |
| ) |
|
|
| demo.launch() |
|
|
