pretrained instead of automodel

app.py

@@ -224,8 +224,8 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
     config.save_pretrained(save_directory=model_loading_name)
     # config = EfficientNetConfig.from_pretrained(model_loading_name)
     
-    # model = EfficientNetPreTrained.from_pretrained(model_loading_name)
-    model = AutoModelForImageClassification.from_pretrained(model_loading_name, trust_remote_code=True)
+    model = EfficientNetPreTrained.from_pretrained(model_loading_name)
+    # model = AutoModelForImageClassification.from_pretrained(model_loading_name, trust_remote_code=True)
     
     # model = EfficientNetPreTrained(config)
     # config.register_for_auto_class()