Tanor commited on
Commit
0e971ba
·
1 Parent(s): 47112e7

changed to BERTic

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from sentiwordnet_calculator import SentimentPipeline
3
 
4
- pipe = SentimentPipeline("Tanor/SRGPTSENTPOS2", "Tanor/SRGPTSENTNEG2")
5
 
6
  def calculate(text):
7
  result = pipe(text)
@@ -20,8 +20,8 @@ iface = gr.Interface(
20
  title="Sentiment Analysis for Serbian",
21
  description="""
22
  This tool performs sentiment analysis on the input text using a model trained on Serbian dictionary definitions.
23
- The model was initially pretrained on the [sr-gpt2-large model by Mihailo Škorić](https://huggingface.co/JeRTeh/sr-gpt2-large),
24
- then fine-tuned on selected definitions from the Serbian WordNet. Please limit the input to 300 tokens.
25
  The outputs represent the Positive (POS), Negative (NEG), and Objective (OBJ) sentiment scores.
26
  """,
27
  examples=[
 
1
  import gradio as gr
2
  from sentiwordnet_calculator import SentimentPipeline
3
 
4
+ pipe = SentimentPipeline("Tanor/BERTicSENTPOS4", "Tanor/BERTicSENTNEG4")
5
 
6
  def calculate(text):
7
  result = pipe(text)
 
20
  title="Sentiment Analysis for Serbian",
21
  description="""
22
  This tool performs sentiment analysis on the input text using a model trained on Serbian dictionary definitions.
23
+ The pretrained model [BERTic from classla](https://huggingface.co/classla/bcms-bertic),
24
+ was fine-tuned on selected definitions from the Serbian WordNet. Please limit the input to 300 tokens.
25
  The outputs represent the Positive (POS), Negative (NEG), and Objective (OBJ) sentiment scores.
26
  """,
27
  examples=[