Update app.py

app.py

@@ -9,7 +9,8 @@ pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, add_special_
 # Definir la función de inferencia
 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)
+    #prediction = pipe(combined)
+    return pipe(combined)
     #return "Comercial" if prediction[0]['label'] == 'LABEL_0' else "Informativo"
 
 # Crear la interfaz de Gradio