using pretrained in register

app.py

@@ -264,7 +264,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     #                                 depth_mult=hparams.depth_mult,)
     
     AutoConfig.register(model_loading_name, EfficientNetConfig)
-    AutoModel.register(EfficientNetConfig, EfficientNet)
+    AutoModel.register(EfficientNetConfig, EfficientNetPreTrained)
     # config = EfficientNetConfig.from_pretrained(model_loading_name)
     
     # model = EfficientNetPreTrained.from_pretrained(model_loading_name)