Update app.py

app.py

@@ -22,7 +22,8 @@ model_list = [
 def predict(chosen_model):
 
     # Initialize the pipeline with the chosen model
-    predicted_label = pipeline("text-classification", model=chosen_model)
+    model_pipeline = pipeline("text-classification", model=chosen_model)
+    predicted_label = model_pipeline[0]
     return predicted_label
 
 inputs = [