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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,8 +17,9 @@ st.markdown('<p style="text-align: center; color: #eaf26f; font-size: 40px; font
17
  authkey = os.getenv('YellowRust')
18
  login(token=authkey)
19
 
 
20
  # Download the model file from Hugging Face
21
- model_path = hf_hub_download(repo_id="shaheer-data/Yellow-Rust-Prediction", filename="final_meta_model.keras")
22
 
23
  # Load the pre-trained model
24
  loaded_model = load_model(model_path)
 
17
  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)