:white_check_mark: Change class name

app.py

@@ -360,7 +360,7 @@ class ReshapedBCEWithLogitsLoss(nn.BCEWithLogitsLoss):
     return super().forward(input.squeeze(), target.float())
 
 
-class DnaBert6MQTLClassifier(nn.Module):
+class DnaBert6MQTLClassifier(nn.Module, PyTorchModelHubMixin):
   def __init__(self,
                seq_len: int, model_repository_name: str,
                bert_model=BertModel.from_pretrained(pretrained_model_name_or_path=DNA_BERT_6),