Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ pipe = TextClassificationPipeline(model=model, tokenizer=tokenizer, add_special_
|
|
| 9 |
# Definir la funci贸n de inferencia
|
| 10 |
def classify_message(ues_detallada, medio_de_comunicacion, asunto_o_copy):
|
| 11 |
combined = f"{ues_detallada} por {medio_de_comunicacion}. {asunto_o_copy}"
|
| 12 |
-
|
| 13 |
# return pipe(combined)
|
| 14 |
return "Comercial" if prediction[0]['label'] == 'LABEL_0' else "Informativo"
|
| 15 |
|
|
|
|
| 9 |
# Definir la funci贸n de inferencia
|
| 10 |
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 |
return "Comercial" if prediction[0]['label'] == 'LABEL_0' else "Informativo"
|
| 15 |
|