Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,8 +11,7 @@ pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, add_special_
|
|
11 |
def classify_message(ues_detallada, medio_de_comunicacion, asunto_o_copy):
|
12 |
combined = f"{ues_detallada.upper()} por {medio_de_comunicacion}. {asunto_o_copy}"
|
13 |
prediction = pipe(combined)
|
14 |
-
|
15 |
-
return {"Comercial" if prediction[0]['label'] == 'LABEL_0' else "Informativo": float(prediction[0]['score']-randoma), "Informativo" if prediction[0]['label'] == 'LABEL_0' else "Comercial": float(1-prediction[0]['score']+randoma)}
|
16 |
|
17 |
# Crear la interfaz de Gradio
|
18 |
iface = gr.Interface(
|
@@ -27,6 +26,6 @@ iface = gr.Interface(
|
|
27 |
description="Describe la unidad de negocio, el canal de comunicaci贸n y el copy de la campa帽a para clasificar si es comercial o informativo."
|
28 |
)
|
29 |
import os
|
30 |
-
Colsubsidio = os.getenv("
|
31 |
# Lanzar la interfaz
|
32 |
-
iface.launch(share = True, auth=("colsubsidio",
|
|
|
11 |
def classify_message(ues_detallada, medio_de_comunicacion, asunto_o_copy):
|
12 |
combined = f"{ues_detallada.upper()} por {medio_de_comunicacion}. {asunto_o_copy}"
|
13 |
prediction = pipe(combined)
|
14 |
+
return "Comercial" if prediction[0]['label'] == 'LABEL_0' else "Informativo"
|
|
|
15 |
|
16 |
# Crear la interfaz de Gradio
|
17 |
iface = gr.Interface(
|
|
|
26 |
description="Describe la unidad de negocio, el canal de comunicaci贸n y el copy de la campa帽a para clasificar si es comercial o informativo."
|
27 |
)
|
28 |
import os
|
29 |
+
Colsubsidio = os.getenv("colsubsidio")
|
30 |
# Lanzar la interfaz
|
31 |
+
iface.launch(share = True, auth=("colsubsidio", colsubsidio))
|