Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from setfit import SetFitModel
 
 # Cargar el modelo
-model = SetFitModel.from_pretrained("desarrolloasesoreslocales/SetFit")
+model = SetFitModel.from_pretrained("desarrolloasesoreslocales/SetFitArgumentos")
 
 # Mapear las etiquetas
-id2label = {0: "World", 1: "Sports", 2: "Business", 3: "Sci/Tech"}
+id2label = {0: "49", 1: "994", 2: "1002", 3: "2014"}
 
 # Definir la función de predicción
 def predict(payload):