Update app.py

app.py

@@ -23,8 +23,9 @@ def predict(chosen_model):
 
     # Initialize the pipeline with the chosen model
     model_pipeline = pipeline("text-classification", model=chosen_model)
-    predicted_label = model_pipeline[0]
-    return predicted_label
+    for prediction in model_pipeline:
+        label = model_pipeline['label']
+    return label
 
 inputs = [
     gr.Textbox(label="Text", value= "As pessoas tem que perceber que ser 'panasca' não é deixar de ser homem, é deixar de ser humano kkk"),