Hugging Face's logo

Spaces:
Vahe
/
FruitClassifier
Runtime error

App Files Community
Vahe commited on
Commit
7965073
·
1 Parent(s): 3069a0a

pkl extension added to model_path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ if system_platform == 'Windows': pathlib.PosixPath = pathlib.WindowsPath
18
  config_manager = ConfigurationManager()
19
  config = config_manager.get_training_config()
20
  MODEL_ROOT = config.trained_model_path
21
- MODEL_NAME = config.params_model_name
22
  MODEL_PATH = os.path.join(MODEL_ROOT, MODEL_NAME)
23
 
24
  def main():
 
18
  config_manager = ConfigurationManager()
19
  config = config_manager.get_training_config()
20
  MODEL_ROOT = config.trained_model_path
21
+ MODEL_NAME = config.params_model_name + '.pkl'
22
  MODEL_PATH = os.path.join(MODEL_ROOT, MODEL_NAME)
23
 
24
  def main():