Spaces:
Sleeping
Sleeping
shaheer-data
commited on
Update app.py
Browse files
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=
|
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)
|