Small fix

script.py

@@ -28,10 +28,12 @@ from data import Dataset
 from util import save_checkpoint, ProgressMeter, AverageMeter, num_params
 from constants import *
 from predict_clickbait import generate_clickbait, tokenizer, classifier_tokenizer
+import os 
 
+os.chdir('naacl-2021-fudge-controlled-generation/')
 
 # imp.reload(model)
-pretrained_model = "./checkpoint-150/"
+pretrained_model = "../checkpoint-150/"
 generation_model = AutoModelForSeq2SeqLM.from_pretrained(pretrained_model, return_dict=True).to(device)
 
 device = 'cuda'
@@ -42,7 +44,7 @@ generation_model.eval()
 model_args = mock.Mock()
 model_args.task = 'clickbait'
 model_args.device = device
-model_args.checkpoint = './checkpoint-1464/'
+model_args.checkpoint = '../checkpoint-1464/'
 
 conditioning_model = Model(model_args, pad_id, len(dataset_info.index2word)) # no need to get the glove embeddings when reloading since they're saved in model ckpt anyway
 conditioning_model = conditioning_model.to(device)