registering directly ENC

app.py

@@ -253,7 +253,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
         # EfficientNetConfig.model_type = "efficientnet_%s_planet_detection" % (hparams.num_channels)
         # EfficientNetConfig.model_type = hparams.model_type
     
-    config.save_pretrained(save_directory=model_loading_name)
+    # config.save_pretrained(save_directory=model_loading_name)
     
     model = EfficientNet(dropout=hparams.dropout,
                                     num_channels=hparams.num_channels,
@@ -263,8 +263,8 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
                                     width_mult=hparams.width_mult,
                                     depth_mult=hparams.depth_mult,)
     
-    AutoConfig.register(model_loading_name, config)
-    AutoModel.register(config, model)
+    AutoConfig.register(model_loading_name, EfficientNetConfig)
+    AutoModel.register(EfficientNetConfig, model)
     # config = EfficientNetConfig.from_pretrained(model_loading_name)
     
     # model = EfficientNetPreTrained.from_pretrained(model_loading_name)