Update README.md

README.md

@@ -39,7 +39,7 @@ session = ort.InferenceSession("resnet18_cifar10.onnx")
 
 # Preprocess image
 def preprocess(img_path):
-    img = Image.open(img_path).convert("RGB").resize((32, 32))
+    img = Image.open(img_path).convert("RGB").resize((224, 224))
     img_data = np.array(img).astype(np.float32) / 255.0
     img_data = np.transpose(img_data, (2, 0, 1))  # CHW format
     img_data = np.expand_dims(img_data, axis=0)   # Batch dimension