Hugging Face's logo

Spaces:
saraayum
/
SARImageColourisation
Sleeping

App Files Community
saraayum commited on
Commit
b6f4361
·
verified ·
1 Parent(s): 740956b

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -9,7 +9,7 @@ from PIL import Image
9
  # ------------------------------
10
  model_path = hf_hub_download(
11
  repo_id="saraayum/sar-colorization-generator", # replace with your actual username/repo
12
- filename="sar_colorization_generator_final.h5"
13
  )
14
  # generator = load_model(model_path, compile=False)
15
  generator = load_model(model_path, compile=False, safe_mode=False)
 
9
  # ------------------------------
10
  model_path = hf_hub_download(
11
  repo_id="saraayum/sar-colorization-generator", # replace with your actual username/repo
12
+ filename="sar_colorization_generator_final.keras"
13
  )
14
  # generator = load_model(model_path, compile=False)
15
  generator = load_model(model_path, compile=False, safe_mode=False)