Hugging Face's logo

Spaces:
tejani
/
Texture2
Paused

App Files Community
tejani commited on
Commit
32400ac
·
verified ·
1 Parent(s): 76c174a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ def process_image(input_path, tile_size=512, seamless=False, use_cpu=False):
27
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
28
 
29
  # Load models
30
- normal_model = load_model(MODELS_DIR / "NormalMapGenerator-CX-Lite_200000_G.pth", use_cpu)
31
- franken_model = load_model(MODELS_DIR / "frankenMapGenerator-CX-Lite_215000_G.pth", use_cpu)
32
 
33
  # Convert to tensor
34
  img_tensor = torch.from_numpy(img.transpose(2, 0, 1)).float() / 255.0
 
27
  img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
28
 
29
  # Load models
30
+ normal_model = load_model(MODELS_DIR / "1x_NormalMapGenerator-CX-Lite_200000_G.pth", use_cpu)
31
+ franken_model = load_model(MODELS_DIR / "1x_frankenMapGenerator-CX-Lite_215000_G.pth", use_cpu)
32
 
33
  # Convert to tensor
34
  img_tensor = torch.from_numpy(img.transpose(2, 0, 1)).float() / 255.0