Update README.md

README.md

@@ -1,6 +1,8 @@
+```py
 from diffusers import AutoPipelineForText2Image
 import torch
 
 pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
 pipeline.load_lora_weights('AntDish/BACK', weight_name='back.safetensors')
-image = pipeline('your prompt').images[0]
+image = pipeline('your prompt').images[0]
+```