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))
|