Update app.py

app.py

@@ -10,8 +10,8 @@ pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, add_special_
 def classify_message(ues_detallada, medio_de_comunicacion, asunto_o_copy):
     combined = f"{ues_detallada} por {medio_de_comunicacion}. {asunto_o_copy}"
     #prediction = pipe(combined)
-    return pipe(combined)
-    #return "Comercial" if prediction[0]['label'] == 'LABEL_0' else "Informativo"
+   # return pipe(combined)
+     return "Comercial" if prediction[0]['label'] == 'LABEL_0' else "Informativo"
 
 # Crear la interfaz de Gradio
 iface = gr.Interface(
@@ -21,7 +21,7 @@ iface = gr.Interface(
         gr.Radio(["EMAIL", "SMS"], label="Canal de Comunicación"),
         gr.Textbox(lines=2, placeholder="Escribe el asunto o copy del mensaje aquí.", label="Asunto o Copy")
     ],
-    outputs= gr.Label(),
+    outputs= gr.Label(label="Tipo de Campaña"),
     title="Clasificador de Campañas",
     description="Describe la unidad de negocio, el canal de comunicación y escribe el copy de la campaña para clasificar si es comercial o informativo."
 )