Spaces:

TeamSAS
/

ImageColorizer

Running

AUMREDKA commited on 2 days ago

Commit

c213867

verified ·

1 Parent(s): cd50def

Expt with checkpoint 22

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 from model import ViTUNetColorizer
 # CKPT = "checkpoints/checkpoint_epoch_015_20250808_154437.pt"
-CKPT = "checkpoints/checkpoint_epoch_024_20250808_200347.pt"
 model = None
 if os.path.exists(CKPT):
     model = ViTUNetColorizer(vit_model_name="vit_tiny_patch16_224").to(device)

 from model import ViTUNetColorizer
 # CKPT = "checkpoints/checkpoint_epoch_015_20250808_154437.pt"
+CKPT = "checkpoints/checkpoint_epoch_022_20250808_190318.pt"
 model = None
 if os.path.exists(CKPT):
     model = ViTUNetColorizer(vit_model_name="vit_tiny_patch16_224").to(device)