diff --git a/function.py b/function.py index 0ca4bfc..f2cbea6 100644 --- a/function.py +++ b/function.py @@ -113,8 +113,8 @@ def img_to_vector(filename): """ with open(filename) as file: - lines = file.readlines() - image_vector = np.array([int(char) for line in lines for char in line.strip()]) + lines = file.readlines() + image_vector = np.array([int(char) for line in lines for char in line.strip()]) return image_vector