Hugging Face's logo

dblasko
/
mirnet-low-light-img-enhancement

Image-to-Image
image-enhancement
computer-vision
Model card Files Community
dblasko commited on
Commit
d504e14
·
1 Parent(s): d77e0b0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -35,10 +35,9 @@ device = (
35
  )
36
 
37
  # Download the model weights from the Hugging Face Hub
38
- model_url = hf_hub_url(
39
- repo_id="dblasko/mirnet-low-light-img-enhancement", filename="mirnet_finetuned.pth" # or mirnet_pretrained.pth
40
  )
41
- model_path = cached_download(model_url)
42
 
43
  # Load the model
44
  model = MIRNet().to(device)
 
35
  )
36
 
37
  # Download the model weights from the Hugging Face Hub
38
+ model_path = hf_hub_download(
39
+ repo_id="dblasko/mirnet-low-light-img-enhancement", filename="mirnet_finetuned.pth"
40
  )
 
41
 
42
  # Load the model
43
  model = MIRNet().to(device)