Hugging Face's logo

Spaces:
clement-bonnet
/
lpn
Running

App Files Community
clement-bonnet commited on
Commit
641e21d
·
1 Parent(s): b6218b2

fix: input blocks

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -23,7 +23,7 @@ from utils import patch_target, ax_to_pil
23
 
24
 
25
  checkpoint_name = "quiet-thunder-789--checkpoint:v0"
26
- BLUE_LOCATION_INPUTS = {0: 13, 1: 9, 2: 9, 3: 6}
27
 
28
  local_dir = snapshot_download(repo_id="clement-bonnet/lpn-2d", allow_patterns=f"{checkpoint_name}/*")
29
  with open(f"{local_dir}/{checkpoint_name}/config.yaml", "r") as f:
 
23
 
24
 
25
  checkpoint_name = "quiet-thunder-789--checkpoint:v0"
26
+ BLUE_LOCATION_INPUTS = {0: 1, 1: 5, 2: 5, 3: 10}
27
 
28
  local_dir = snapshot_download(repo_id="clement-bonnet/lpn-2d", allow_patterns=f"{checkpoint_name}/*")
29
  with open(f"{local_dir}/{checkpoint_name}/config.yaml", "r") as f: