Update model.py

model.py

@@ -335,8 +335,8 @@ class MultiHeadClassification(nn.Module):
             dropout (float): Dropout rate
             l2_reg (float): L2 regularization rate
         """
-        backbone = AutoModel.from_pretrained(os.path.join(model_path, 'pretrained/backbone.pth'))
+        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/model.pth'))
+        instance.load(os.path.join(model_path, 'pretrained/multi-head-sequence-classification-model-model.pth'))
         instance.head_config = {k: v. instance.heads}
         return instance