Update product_return_prediction/api.py

product_return_prediction/api.py

@@ -160,9 +160,9 @@ async def predict(products: ProductRequest):
     # download_file(f"{models_uri}{model_name}", model_path, headers)
     # download_file(f"{models_uri}{scaler_name}", scaler_path, headers)
 
-    # model = load_model(model_path)
+    model = load_model(hf_hub_download(repo_id="molinari135/se4ai-models", filename="svm.pkl"))
 
-    model = pickle.load(hf_hub_download(repo_id="molinari135/se4ai-models", filename="svm.pkl"))
+    # model = pickle.load(hf_hub_download(repo_id="molinari135/se4ai-models", filename="svm.pkl"))
     
     scaled_inventory = apply_scaling(prepared_inventory, scaler_file)
     predictions, probabilities = make_predictions(model, scaled_inventory)