try fix test error

divakaivan · divakaivan · commit 3f80b1c4137a · 2025-02-09T16:40:48.000+09:00
diff --git a/src/make_api/app/main.py b/src/make_api/app/main.py
@@ -13,11 +13,11 @@
 
 
 @lru_cache
-def get_model():
+def get_model(model_uri):
     return mlflow.sklearn.load_model(model_uri)
 
 
-pipe = get_model()
+pipe = get_model(model_uri)
 
 app = FastAPI()
 
