From b3e7203cadc8b95e2e8ec00b8ccc1ffa5825cf97 Mon Sep 17 00:00:00 2001
From: Shivansh Mittal <88429611+Shivansh20128@users.noreply.github.com>
Date: Thu, 12 Dec 2024 00:58:34 +0530
Subject: [PATCH] Adding image size in data loader in README (#151)

adding image size in data loader
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 764239b..0b3a60f 100644
--- a/README.md
+++ b/README.md
@@ -80,7 +80,7 @@ from piqture.data_loader.mnist_data_loader import load_mnist_dataset
 from piqture.embeddings.image_embeddings.ineqr import INEQR
 
 # Load MNIST dataset
-train_dataset, test_dataset = load_mnist_dataset()
+train_dataset, test_dataset = load_mnist_dataset(img_size=(8, 8))
 
 # Retrieve a single image from the dataset
 image, label = train_dataset[0]