From e89cd155ae33295794062de07db42446f2656f04 Mon Sep 17 00:00:00 2001
From: Wojciech Jarmosz <wojciech@jarmosz.com>
Date: Sun, 25 Apr 2021 22:19:56 +0200
Subject: [PATCH] Fix folder load path

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

diff --git a/linear_regression.py b/linear_regression.py
index 3364cb9..3afb052 100644
--- a/linear_regression.py
+++ b/linear_regression.py
@@ -23,7 +23,7 @@ test_Y = tf.convert_to_tensor(test_data[["IMDb"]])
 normalizer = preprocessing.Normalization(input_shape=[3,])
 normalizer.adapt(train_X)
 
-if os.path.isfile('linear_regression.h5'):
+if os.path.isdir('linear_regression'):
     model = keras.models.load_model('linear_regression')
 else:
     model = keras.Sequential([