Update README.md

README.md

@@ -63,7 +63,7 @@ Then to forward an image:
 
 
 ```python
-img = np.random.rand(-1024, 1024, size=(256, 256)) # single axial slice, in PL orientation
+img = 2048 * np.random.rand(256, 256) - 1024 # single axial slice, in PL orientation
 model_input = processor(img)
 features = model(**model_input)
 ```