Update product_return_prediction/api.py
Browse files
product_return_prediction/api.py
CHANGED
@@ -150,8 +150,8 @@ async def predict(products: ProductRequest):
|
|
150 |
|
151 |
headers = f"Authorization: Bearer {hf_token}"
|
152 |
|
153 |
-
download_file(f{models_uri}{model_name}, model_path, headers)
|
154 |
-
download_file(f{models_uri}{scaler_name}, scaler_path, headers)
|
155 |
|
156 |
model = load_model(model_path)
|
157 |
|
|
|
150 |
|
151 |
headers = f"Authorization: Bearer {hf_token}"
|
152 |
|
153 |
+
download_file(f"{models_uri}{model_name}", model_path, headers)
|
154 |
+
download_file(f"{models_uri}{scaler_name}", scaler_path, headers)
|
155 |
|
156 |
model = load_model(model_path)
|
157 |
|