Update app.py

app.py

@@ -23,8 +23,7 @@ def predict(chosen_model):
 
     # Initialize the pipeline with the chosen model
     model_pipeline = pipeline("text-classification", model=chosen_model)
-    for prediction in model_pipeline:
-        label = model_pipeline['label']
+    label = model_pipeline[0]['label']
     return label
 
 inputs = [