inference_example.py

#!/usr/bin/env python3
"""
Bengali Sentiment Analysis - Inference Example
===============================================
Usage: python inference_example.py
"""

from transformers import pipeline

MODEL = "OsamaBinLikhon/bengali-code-mix-sentiment"

def analyze(text):
    """Analyze sentiment of Bengali/code-mixed text."""
    classifier = pipeline("sentiment-analysis", model=MODEL)
    return classifier(text[:512])[0]

if __name__ == "__main__":
    samples = [
        "আজকের দিনটা খুবই ভালো গেছে!",
        "এই প্রোডাক্টটা একদম খারাপ",
    ]
    
    for text in samples:
        print(f"Text: {text}")
        print(f"Sentiment: {analyze(text)}
")