registering EN directly

app.py

@@ -255,16 +255,16 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     
     # config.save_pretrained(save_directory=model_loading_name)
     
-    model = EfficientNet(dropout=hparams.dropout,
-                                    num_channels=hparams.num_channels,
-                                    num_classes=hparams.num_classes,
-                                    size=hparams.size,
-                                    stochastic_depth_prob=hparams.stochastic_depth_prob,
-                                    width_mult=hparams.width_mult,
-                                    depth_mult=hparams.depth_mult,)
+    # model = EfficientNet(dropout=hparams.dropout,
+    #                                 num_channels=hparams.num_channels,
+    #                                 num_classes=hparams.num_classes,
+    #                                 size=hparams.size,
+    #                                 stochastic_depth_prob=hparams.stochastic_depth_prob,
+    #                                 width_mult=hparams.width_mult,
+    #                                 depth_mult=hparams.depth_mult,)
     
     AutoConfig.register(model_loading_name, EfficientNetConfig)
-    AutoModel.register(EfficientNetConfig, model)
+    AutoModel.register(EfficientNetConfig, EfficientNet)
     # config = EfficientNetConfig.from_pretrained(model_loading_name)
     
     # model = EfficientNetPreTrained.from_pretrained(model_loading_name)