fix error for lp and contr

.gitignore

@@ -0,0 +1 @@
+Pipfile*

app.py

@@ -56,12 +56,12 @@ def ssl_predict(in_text, model_type):
 
     elif model_type == "contrastive_reg":
         model = FixMatchTune(encoder_name="readerbench/RoBERT-base")
-        model.cls_head.load_weights("./checkpoints/contrastive")
+        model.load_weights("./checkpoints/contrastive")
         preds, _ = model([toks["input_ids"],toks["attention_mask"]], training=False)
 
     elif model_type == "label_propagation":
         model = LPModel()
-        model.cls_head.load_weights("./checkpoints/label_prop")
+        model.load_weights("./checkpoints/label_prop")
         preds = model([toks["input_ids"],toks["attention_mask"]], training=False)
 
     probs = list(preds[0].numpy())