|
import gradio as gr |
|
from sentiwordnet_calculator import SentimentPipeline |
|
|
|
pipe = SentimentPipeline("Tanor/SRGPTSENTPOS4", "Tanor/SRGPTSENTNEG4") |
|
|
|
|
|
def calculate(text): |
|
result = pipe(text) |
|
|
|
visual = result |
|
|
|
numerical = {key: round(value, 2) for key, value in result.items()} |
|
|
|
numerical_str = ", ".join(f"{key}: {value}" for key, value in numerical.items()) |
|
return visual, numerical_str |
|
|
|
iface = gr.Interface( |
|
fn=calculate, |
|
inputs=gr.inputs.Textbox(lines=5, placeholder="Enter your text here..."), |
|
outputs=[gr.outputs.Label(num_top_classes=3), "text"], |
|
title="Sentiment Analysis for Serbian", |
|
description=""" |
|
Ovaj alat vrši analizu sentimenta na uneseni tekst koristeći model obučen na definicijama srpskog rečnika. |
|
Model je prvobitno bio prethodno obučen na <a href='https://huggingface.co/JeRTeh/sr-gpt2-large' target='_blank'>sr-gpt2-large modelu Mihaila Škorića</a>, |
|
a potom je bio usavršen na odabranim definicijama iz srpskog WordNeta. Molimo vas da ograničite unos na 300 tokena. |
|
Izlazi predstavljaju pozitivne (POS), negativne (NEG) i objektivne (OBJ) ocene sentimenta. |
|
|
|
This tool performs sentiment analysis on the input text using a model trained on Serbian dictionary definitions. |
|
The model was initially pretrained on the <a href='https://huggingface.co/JeRTeh/sr-gpt2-large' target='_blank'>sr-gpt2-large model by Mihailo Škorić</a>, |
|
then fine-tuned on selected definitions from the Serbian WordNet. Please limit the input to 300 tokens. |
|
The outputs represent the Positive (POS), Negative (NEG), and Objective (OBJ) sentiment scores. |
|
""", |
|
, |
|
, |
|
examples=[ |
|
["osoba koja ne prihvata nove ideje"], |
|
["intenzivna ojađenost"], |
|
["uopštenih osećanja tuge"], |
|
["žalostan zbog gubitka ili uskraćenosti"], |
|
["činjenje dobra; osećaj dobrotvornosti"], |
|
["Jako pozitivno osećanje poštovanja i privrežen..."], |
|
["usrećiti ili zadovoljiti"], |
|
["Korisna ili vredna osobina"], |
|
["uzrokovati strah"], |
|
["svojstvo onoga kome nedostaje živost"], |
|
["koji čini živahnim i veselim"], |
|
["Biti zdrav, osećati se dobro."], |
|
|
|
] |
|
|
|
) |
|
|
|
iface.launch() |
|
|
