Update app.py

app.py

@@ -18,7 +18,7 @@ model_link = hf_hub_download(repo_id="FFZG-cleopatra/Croatian-News-Classifier",f
 multitask_model.load_state_dict(torch.load(model_link, map_location=device))
 multitask_model.to(device)
 
-def predict(sentence = "Volim ti"):
+def predict_sentiment(sentence = "Volim ti"):
     # gather everyone if you want to have a single DatasetDict
     document = DatasetDict({
         # "train": Dataset.from_pandas(df_document_sl_hr_train),
@@ -65,7 +65,7 @@ def predict(sentence = "Volim ti"):
 
 
 interface = gr.Interface(
-    fn=get_sentiment,
+    fn=predict_sentiment,
     inputs='text',
     outputs=['text', 'label'],
     title='Sentiment Analysis',