From 343ebbf3077013c0ec915fd0a3ed6363710e6ae2 Mon Sep 17 00:00:00 2001
From: Sheaza <nataliawolowiec3@gmail.com>
Date: Sun, 9 Jun 2024 18:05:59 +0200
Subject: [PATCH] sacred fix 1

---
 train_sacred.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_sacred.py b/train_sacred.py
index 451c0e5..06a74d5 100644
--- a/train_sacred.py
+++ b/train_sacred.py
@@ -54,7 +54,7 @@ class RegressionModel:
 
     def save_model(self):
         self.model.save("model.keras")
-        ex.add_artifact("model.keras")
+        ex.add_artifact("model.keras", content_type="application/octet-stream")
 
 
 @ex.main