got rid of config when loading pretrained model
Browse files
app.py
CHANGED
@@ -223,7 +223,7 @@ def predict_and_analyze(model_name, num_channels, dim, input_channel, image):
|
|
223 |
config.save_pretrained(save_directory=model_loading_name)
|
224 |
# config = EfficientNetConfig.from_pretrained(model_loading_name)
|
225 |
|
226 |
-
model = EfficientNetPreTrained.from_pretrained(model_loading_name
|
227 |
|
228 |
# model = EfficientNetPreTrained(config)
|
229 |
# config.register_for_auto_class()
|
|
|
223 |
config.save_pretrained(save_directory=model_loading_name)
|
224 |
# config = EfficientNetConfig.from_pretrained(model_loading_name)
|
225 |
|
226 |
+
model = EfficientNetPreTrained.from_pretrained(model_loading_name)
|
227 |
|
228 |
# model = EfficientNetPreTrained(config)
|
229 |
# config.register_for_auto_class()
|