Spaces:

johnberg
/

CLIPInverter

Build error

johnberg commited on Jul 26, 2023

Commit

0bcc919

1 Parent(s): b88e502

update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,8 +32,8 @@ input_transforms = transforms.Compose([
             transforms.ToTensor(),
             transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
-model_path = "/scratch/users/aanees20/hpc_run/light_textmod/textmodulation/mapper/exp_ada5/checkpoints/best_model.pt"
-e4e_path = 'e4e_ffhq_encode.pt' # "/scratch/users/abaykal20/hpc_run/sam/SAM/pretrained_models/e4e_ffhq_encode.pt"
 ckpt = torch.load(model_path, map_location='cpu')
 opts = ckpt['opts']

             transforms.ToTensor(),
             transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
+model_path = 'pretrained_faces.pt'
+e4e_path = 'e4e_ffhq_encode.pt'
 ckpt = torch.load(model_path, map_location='cpu')
 opts = ckpt['opts']