Update README.md

README.md

@@ -61,12 +61,12 @@ To colorize an image, follow these steps:
 
     # Set the path to your image
     image_path = 'path_to_your_image.jpg'
-    save_path = 'path_to_save_colorized_image'
+    save_dir = 'dir_to_save_colorized_image'
     criterion = nn.MSELoss()
 
     # Colorize the image
     use_gpu = torch.cuda.is_available()
-    colorize_single_image(image_path, model, criterion, save_path, epoch=0, use_gpu=use_gpu)
+    colorize_single_image(image_path, model, criterion, save_dir, epoch=0, use_gpu=use_gpu)
     ```
 
 3. **View the Results**: The colorized image will be saved in the specified `save_path`. You will find both the original grayscale and the colorized version there.