deb

src/model_utils.py

@@ -135,6 +135,8 @@ def save_model(model, user_model_name, metrics_result_single=None):
         path_or_fileobj=buf
     )
 
+    print('filename SAVEEEEEEEEEEEEE', filename)
+    st.warning(f'SAVEEEEEEEEEEEEE {filename}')
     return filename
 
 # def save_model_ensemble(models, user_model_name, best_iterations=None, fold_scores=None, metrics_result_ensemble=None):
@@ -223,6 +225,8 @@ def save_model_ensemble(models, user_model_name, best_iterations=None, fold_scor
         path_or_fileobj=buf
     )
 
+    print('filename SAVEEEEEEEEEEEEEEEE', filename)
+    st.warning(f'SAVEEEEEEEEEEEEEEEEEEE {filename}')
     return filename
 
 
@@ -282,6 +286,10 @@ def load_model(model_name):
         raise FileNotFoundError(f"Model {model_name} not found in repo.")
 
     model_bytes = row["model_file"]["bytes"]
+
+    print("LOADDDDDDDDDDDDDDDDDDDDDDDDDD")
+    print('row["filename"]', row["filename"])
+    print('model_name.replace("parquet", "pkl")', model_name.replace("parquet", "pkl"))
     return pickle.loads(model_bytes)