Upload model.py with huggingface_hub

model.py

@@ -11,7 +11,7 @@ from transformers import PretrainedConfig
 ######################################################################
 class VesselSegmentConfig(PretrainedConfig):
   model_type = "SegformerForSemanticSegmentation"
-  def __init__(self, num_classes, input_channels, image_size, features, attention_dims, patch_size, batch_size, **kwargs):
+  def __init__(self, num_classes=1, input_channels=1, image_size=[512,512], features=[64,64,128], attention_dims=[64,32,16], patch_size=256, batch_size=4, **kwargs):
     self.num_classes = num_classes
     self.input_channels = input_channels
     self.image_size = image_size