|
import gradio as gr |
|
from transformers import pipeline |
|
|
|
|
|
model = pipeline(task="sentiment-analysis", model="skl25/banglabert-sentiment") |
|
|
|
|
|
def predict_sentiment(text): |
|
result = model(text) |
|
sentiment = result[0]['label'] |
|
if sentiment == 1: |
|
return "Positive" |
|
elif sentiment == "2": |
|
return "Negative" |
|
else: |
|
return "Neutral" |
|
|
|
|
|
predefined_examples = [ |
|
"গবেষণার কাজে আমাদের সাহায্য করার জন্য রাইজকে অসংখ্য ধন্যবাদ", |
|
"আমি বুয়েট এনএলপি এর হাগিং ফেস মডেলগুলি ব্যবহার করতে ভালোবাসি!", |
|
"এই বইটি বিজ্ঞান ও প্রযুক্তির সমৃদ্ধি সম্পর্কে বিস্তারিত জ্ঞান প্রদান করে।", |
|
"আমি খুব কষ্ট পাচ্ছি", "মডেলটি সঠিকভাবে অনুভূতি ধরতে ব্যর্থ হয়েছে।", |
|
"আমার জীবন এখন একটি অস্বাভাবিক দুঃখে ভরা আছে।", |
|
"এই বিষয়ে আমার কোন অভিজ্ঞতা নেই।", |
|
"আমি এই ব্যাপারে কিছুই জানি না", |
|
"আমি এই বইটি ভালোভাবে পড়তে পাচ্ছি", |
|
"এটি একটি অসাধারণ অভিজ্ঞতা ছিল", |
|
"আমার মনে হয় এই ফিল্মটি পুরস্কৃত হওয়া উচিত" |
|
] |
|
|
|
|
|
iface = gr.Interface( |
|
fn=predict_sentiment, |
|
inputs=gr.Textbox(label="Text"), |
|
outputs="text", |
|
live=False, |
|
examples=predefined_examples, |
|
title="Bangla Sentiment Analysis", |
|
theme='abidlabs/pakistan' |
|
) |
|
|
|
|
|
iface.launch(inline=False) |
|
|