sravan837 commited on
Commit
fddfdf3
·
1 Parent(s): 821dcee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -12,9 +12,13 @@ app = FastAPI()
12
  # Define Hugging Face repo details
13
  HF_REPO_ID = "sravan837/slope-stability-model"
14
 
15
- # Download the preprocessor and model from Hugging Face
16
- onnx_model_path = hf_hub_download(repo_id=HF_REPO_ID, filename="slope_stability_model.onnx")
17
 
 
 
 
 
 
 
18
  # Load preprocessor
19
  preprocessor = joblib.load(preprocessor_path)
20
 
 
12
  # Define Hugging Face repo details
13
  HF_REPO_ID = "sravan837/slope-stability-model"
14
 
 
 
15
 
16
+ # Download the preprocessor and model from Hugging Face
17
+ onnx_model_path = hf_hub_download(
18
+ repo_id="sravan837/slope-stability-model",
19
+ filename="slope_stability_model.onnx",
20
+ token="sravan837"
21
+ )
22
  # Load preprocessor
23
  preprocessor = joblib.load(preprocessor_path)
24