Spaces:

TeamSAS
/

ImageColorizer

Sleeping

AUMREDKA commited on Aug 11

Commit

1dc3825

verified ·

1 Parent(s): 9b0ef83

Use latest model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ from model import ViTUNetColorizer
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-CKPT = "checkpoints/checkpoint_epoch_015_20250808_154437.pt"
 model = None
 if os.path.exists(CKPT):
     print(f"Loading model from: {CKPT}")

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+CKPT = "checkpoints/checkpoint_epoch_017_20250810_193435.pt"
 model = None
 if os.path.exists(CKPT):
     print(f"Loading model from: {CKPT}")