actualizando el app

app.py

@@ -61,15 +61,11 @@ def load():
     path_images = 'data/doc_explore/DocExplore_images/'
     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"
 
-    try:
-        model = Model().to(device)
-        model_checkpoint = torch.load(path_model) # 'model_60k_images_073.ckpt' -> modelo entrenado con 60k imagenes sin pidinet
-        model.load_state_dict(model_checkpoint['state_dict'])                # 'modified_model_083.ckpt' -> modelo entrenado con 60k imagenes con pidinet
-        model.eval()                                                         # 'original_model_083.ckpt' -> modelo original entrenado con 60k imagenes con pidinet
-        print("Modelo cargado exitosamente")
-    except:
-        print("No se pudo cargar el modelo. Intenta nuevamente cambiando el argumento --model_type")
-        exit()
+    
+    model_checkpoint = torch.load(path_model) # 'model_60k_images_073.ckpt' -> modelo entrenado con 60k imagenes sin pidinet
+    model.load_state_dict(model_checkpoint['state_dict'])                # 'modified_model_083.ckpt' -> modelo entrenado con 60k imagenes con pidinet
+    model.eval()                                                         # 'original_model_083.ckpt' -> modelo original entrenado con 60k imagenes con pidinet
+    print("Modelo cargado exitosamente")
 
     embeddings_file_1 = hf_hub_download(repo_id="CHSTR/DocExplore", filename="dino_flicker_docexplore_groundingDINO.pkl")
     embeddings_file_0 = hf_hub_download(repo_id="CHSTR/DocExplore", filename="docexp_embeddings.pkl")