Update app.py

app.py

@@ -26,16 +26,23 @@ model = SetFitModel.from_pretrained("desarrolloasesoreslocales/SetFitPruebaRecor
 
 # Definir la función de predicción
 def predict(payload):
-    recorte = ""
+    
+    recorte_general = ""
+
+    for chunk in cortar_en_bloques(s, 90):
+      if model.predict([chunk]).item() == 1:
+        recorte_general += chunk + " "
+
+    recorte_final = ""
 
-    for chunk in cortar_en_bloques(payload, 100):
+    for chunk in cortar_en_bloques(recorte_general, 84):
       if model.predict([chunk]).item() == 1:
-        recorte += chunk
+        recorte_final += chunk + " "
           
     # results = model.predict_proba([payload])[0]
     # return "\n".join([f"Label: {id2label[i]},      Score: {round(result.item()*100, 2)} %" for i, result in enumerate(results)])
     
-    return recorte
+    return recorte_final
 
 # Crear una interfaz Gradio
 iface = gr.Interface(