Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ def classify_message(ues_detallada, medio_de_comunicacion, asunto_o_copy):
|
|
11 |
combined = f"{ues_detallada} por {medio_de_comunicacion}. {asunto_o_copy}"
|
12 |
#prediction = pipe(combined)
|
13 |
# return pipe(combined)
|
14 |
-
|
15 |
|
16 |
# Crear la interfaz de Gradio
|
17 |
iface = gr.Interface(
|
|
|
11 |
combined = f"{ues_detallada} por {medio_de_comunicacion}. {asunto_o_copy}"
|
12 |
#prediction = pipe(combined)
|
13 |
# return pipe(combined)
|
14 |
+
return "Comercial" if prediction[0]['label'] == 'LABEL_0' else "Informativo"
|
15 |
|
16 |
# Crear la interfaz de Gradio
|
17 |
iface = gr.Interface(
|