Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
+from transformers import TextClassificationPipeline
+
+pipe = TextClassificationPipeline(model="hedtorresca/Multilingual-MiniLM-L12-H384-fine-tunning2")
 
 # Definir la función de inferencia
 def classify_message(ues_detallada, medio_de_comunicacion, asunto_o_copy):
     combined = f"{ues_detallada} por {medio_de_comunicacion}. {asunto_o_copy}"
-    prediction = pipe(combined, model="models/hedtorresca/Multilingual-MiniLM-L12-H384-fine-tunning2")
+    prediction = pipe(combined)
     return "Comercial" if prediction[0]['label'] == 'LABEL_0' else "Informativo"
 
 # Crear la interfaz de Gradio