Update app.py

app.py

@@ -30,7 +30,7 @@ def predict_sentiment(text_input, model_selection):
         predicted_class = result["labels"][0]
         inference_time = end_time - start_time
         model_size = pretrained_tokenizer.model_max_length + pretrained_model.model.num_parameters()
-        return candidate_labels[predicted_class], inference_time, model_size
+        return predicted_class, inference_time, model_size
 
 inputs = [
     gr.inputs.Textbox("Enter text"),