Update README.md

README.md

@@ -13,8 +13,11 @@ license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICE
 # Flux_finetune_with_swami_Vivekananda final
 
 <Gallery />
+
 ## Inference
 
+```python
+
 import torch
 from diffusers import FluxPipeline
 
@@ -28,9 +31,10 @@ prompt = "A young Indian monk with sunglasses in a cricket ground, holding a bat
 image = pipe(prompt, 
              num_inference_steps=24, 
              guidance_scale=3.5,
-             width=512, height=512,
+             width=960, height=1280,
             ).images[0]
 image.save(f"example.png")
+```
 
 ## Model description