actualizando el app

app.py

@@ -62,7 +62,7 @@ def load():
     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_checkpoint = torch.load(path_model) # 'model_60k_images_073.ckpt' -> modelo entrenado con 60k imagenes sin pidinet
+    model_checkpoint = torch.load(path_model, map_location=device) # '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")