Hugging Face's logo

Spaces:
TeamSAS
/
ImageColorizer
Running

App Files Community
AUMREDKA commited on
Commit
1dc3825
·
verified ·
1 Parent(s): 9b0ef83

Use latest model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ from model import ViTUNetColorizer
20
 
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
22
 
23
- CKPT = "checkpoints/checkpoint_epoch_015_20250808_154437.pt"
24
  model = None
25
  if os.path.exists(CKPT):
26
  print(f"Loading model from: {CKPT}")
 
20
 
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
22
 
23
+ CKPT = "checkpoints/checkpoint_epoch_017_20250810_193435.pt"
24
  model = None
25
  if os.path.exists(CKPT):
26
  print(f"Loading model from: {CKPT}")