Update README.md

README.md

@@ -51,14 +51,14 @@ gray_inpaintor = AutoModel.from_pretrained(
     trust_remote_code=True, 
 )
 
-Load the gray2rgb model
+# Load the gray2rgb model
 gray2rgb = AutoModel.from_pretrained(
     'jwengr/stable-diffusion-2-gray-inpaint-to-rgb', 
     subfolder='gray2rgb', 
     trust_remote_code=True, 
 )
 
-Move models to GPU
+# Move models to GPU
 gray_inpaintor.to('cuda')
 gray2rgb.to('cuda')