actualizando el app

app.py

@@ -24,6 +24,7 @@ import os
 # No reconoce la carpeta que esta dos niveles abajo src
 from src.model_LN_prompt import Model
 from src.options import opts
+from datasets import load_dataset
 
 
 DEBUG = False
@@ -58,7 +59,12 @@ else:
 
 @st.cache_resource
 def load():
-    path_images = 'data/doc_explore/DocExplore_images/'
+    # Descargamos el dataset
+    dataset = load_dataset("CHSTR/doc_explore")
+    local_dir = "./"
+    dataset.save_to_disk(local_dir)
+
+    path_images = 'data/'
     path_model = hf_hub_download(repo_id="CHSTR/DocExplore", filename="epoch=16-mAP=0.66_triplet.ckpt")#"models/epoch=16-mAP=0.66_triplet.ckpt"
 
     model = Model()