Spaces:

TeamSAS
/

ImageColorizer

Running

AUMREDKA commited on 3 days ago

Commit

f7c8e01

verified ·

1 Parent(s): 3333da8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,8 +17,7 @@ from skimage.metrics import structural_similarity as ssim_metric
 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_022_20250808_190318.pt"
 model = None
 if os.path.exists(CKPT):
     model = ViTUNetColorizer(vit_model_name="vit_tiny_patch16_224").to(device)

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