Update app.py

app.py

@@ -56,10 +56,11 @@ def predict_sentiment(sentence = "Volim ti"):
                 torch.unsqueeze(batch["attention_mask"], 0),)
         
         print(tokenizer.decode(batch["input_ids"],skip_special_tokens=True))
+        print("logits:",classifier_output.logits)
         prediction =torch.max(classifier_output.logits, axis=1)
         predictions.append(prediction.indices.item())
     
-    print("p:", predictions[0] , id2label[predictions[0]] )
+    print("predictions:", predictions[0] , id2label[predictions[0]] )
     return id2label[predictions[0]]
 
 
@@ -67,7 +68,7 @@ interface = gr.Interface(
     fn=predict_sentiment,
     inputs='text',
     outputs=['label'],
-    title='Sentiment Analysis',
+    title='Croatian News Sentiment Analysis 1.0',
     description='Get the positive/neutral/negative sentiment for the given input.'
 )