Upload ./example.py with huggingface_hub

example.py

@@ -2,9 +2,9 @@ from model import BrainEncodingModel
 from config_utils import load_from_yaml
 import torch
 
-cfg = load_from_yaml("/workspace/model_packed/config.yaml")
+cfg = load_from_yaml("./config.yaml")
 model = BrainEncodingModel(cfg)
-sd_path = '/nfscc/alg23/xalex_ab/all/t78256_00001_DATASET.ROIS=all,/soup_newnew.pth'
+sd_path = './ckpt.pth'
 sd = torch.load(sd_path)
 model.load_state_dict(sd)
 model.eval().cuda()
@@ -18,6 +18,7 @@ def transform_image(x):
 x = transform_image(x)
 x = x.cuda()
     
+    
 subject = 'subj01'  # could be 1 of 8 subjects
 
 with torch.no_grad():