Update classify.py

mpcs-52040 · Oct 11, 2024 · be1b86c · be1b86c
1 parent f04fdaf
commit be1b86c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/part1/classify.py b/part1/classify.py
@@ -20,7 +20,7 @@ def load_image(path):
     parser = argparse.ArgumentParser()
     parser.add_argument('--input', type=str, required=True)
     parser.add_argument('--model-path', type=str,
-                        required=False, default='/tmp/mnist_model/')
+                        required=False, default='/tmp/mnist_model/mnist_model.keras')
     args = parser.parse_args()
 
     print('Loading model from:', args.model_path)