jpterry commited on
Commit
9907744
·
1 Parent(s): 009d5db

got rid of config when loading pretrained model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 + '/config.json')
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()