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