Update moo.py

moo.py

@@ -47,11 +47,11 @@ if 'Half-Life' in args.objectives:
     halflife_model = HalfLifeModel(device=device)
     score_models.append(halflife_model)
 if 'Affinity' in args.objectives:
-    affinity_predictor = load_affinity_predictor('/scratch/pranamlab/tong/checkpoints/MOG-DFM/classifier_ckpt/binding_affinity_unpooled.pt', device)
+    affinity_predictor = load_affinity_predictor('./classifier_ckpt/binding_affinity_unpooled.pt', device)
     affinity_model = AffinityModel(affinity_predictor, target_sequence)
     score_models.append(affinity_model)
 if 'Motif' in args.objectives or 'Specificity' in args.objectives:
-    bindevaluator = load_bindevaluator('/scratch/pranamlab/tong/checkpoints/BindEvaluator/model_path/finetuned_BindEvaluator.ckpt', device)
+    bindevaluator = load_bindevaluator('./classifier_ckpt/finetuned_BindEvaluator.ckpt', device)
     motif_model = MotifModel(bindevaluator, target_sequence, motifs, penalty=args.motif_penalty)
     score_models.append(motif_model)