CHSTR commited on
Commit
afdbdc8
1 Parent(s): fec6d6e

actualizando el app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def load():
61
  path_images = 'data/doc_explore/DocExplore_images/'
62
  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"
63
 
64
-
65
  model_checkpoint = torch.load(path_model, map_location=device) # 'model_60k_images_073.ckpt' -> modelo entrenado con 60k imagenes sin pidinet
66
  model.load_state_dict(model_checkpoint['state_dict']) # 'modified_model_083.ckpt' -> modelo entrenado con 60k imagenes con pidinet
67
  model.eval() # 'original_model_083.ckpt' -> modelo original entrenado con 60k imagenes con pidinet
 
61
  path_images = 'data/doc_explore/DocExplore_images/'
62
  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"
63
 
64
+ model = Model()
65
  model_checkpoint = torch.load(path_model, map_location=device) # 'model_60k_images_073.ckpt' -> modelo entrenado con 60k imagenes sin pidinet
66
  model.load_state_dict(model_checkpoint['state_dict']) # 'modified_model_083.ckpt' -> modelo entrenado con 60k imagenes con pidinet
67
  model.eval() # 'original_model_083.ckpt' -> modelo original entrenado con 60k imagenes con pidinet