shaheer-data commited on
Commit
b0bc16e
·
verified ·
1 Parent(s): 512099d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,8 +18,9 @@ authkey = os.getenv('YellowRust')
18
  login(token=authkey)
19
 
20
  model_id = os.getenv('model_repo_id')
 
21
  # Download the model file from Hugging Face
22
- model_path = hf_hub_download(repo_id=model_id, filename="final_meta_model.keras")
23
 
24
  # Load the pre-trained model
25
  loaded_model = load_model(model_path)
 
18
  login(token=authkey)
19
 
20
  model_id = os.getenv('model_repo_id')
21
+ model_file = os.getenv('model_path')
22
  # Download the model file from Hugging Face
23
+ model_path = hf_hub_download(repo_id=model_id, filename=model_file)
24
 
25
  # Load the pre-trained model
26
  loaded_model = load_model(model_path)