Update model.py

model.py

@@ -337,6 +337,6 @@ class MultiHeadClassification(nn.Module):
         """
         backbone = AutoModel.from_pretrained(os.path.join(model_path, 'pretrained/backbone'))
         instance = cls(backbone, head_config, dropout, l2_reg)
-        instance.load(os.path.join(model_path, 'pretrained/multi-head-sequence-classification-model-model.pth'))
+        instance.load(os.path.join(model_path, 'multi-head-sequence-classification-model-model.pth'))
         instance.head_config = {k: v. instance.heads}
         return instance