hedtorresca commited on
Commit
fd883f9
verified
1 Parent(s): 971808c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
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
- randoma = random.uniform(0, 0.4)
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("Colsubsidio")
31
  # Lanzar la interfaz
32
- iface.launch(share = True, auth=("colsubsidio", 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))