Update classify.py

mpcs-52040 · Oct 11, 2024 · 5d1aed6 · 5d1aed6
1 parent c1bcec1
commit 5d1aed6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/part1/classify.py b/part1/classify.py
@@ -24,7 +24,7 @@ def load_image(path):
     args = parser.parse_args()
 
     print('Loading model from:', args.model_path)
-    model = tf.saved_model.load(args.model_path)
+    model = tf.keras.models.load_model(args.model_path)
 
     print('Loading image from:', args.input)
     img = load_image(args.input)