Update app.py

app.py

@@ -31,12 +31,11 @@ pipeline
 
 
 
-# Question answering pipeline, specifying the checkpoint identifier
-# model = AutoModelForSequenceClassification.from_pretrained(
-#     pretrained_model_name_or_path= "thak123/Cro-Frida",
-#     num_labels=3,
-# )
-
+label2id =  {
+    "LABEL_0": "negative",
+    "LABEL_1": "neutral",
+    "LABEL_2": "positive"
+  }
 
 analyzer = pipeline(
 
@@ -44,10 +43,7 @@ analyzer = pipeline(
 
 )
 def predict_sentiment(x):
-    return analyzer(x)
-
-
-
+    return label2id[analyzer(x)[0]["label"]]
 
 
 interface = gr.Interface(
@@ -55,7 +51,7 @@ interface = gr.Interface(
     inputs='text',
     outputs=['text'],
     title='Croatian Movie reviews Sentiment Analysis',
-    examples= ["Es mīlu Tevi","Es ienīstu kafiju"],
+    examples= ["Volim kavu","Ne volim kavu"],
     description='Get the positive/neutral/negative sentiment for the given input.'
 )