Update README.md

README.md

@@ -13,7 +13,24 @@ license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICE
 # Flux_finetune_with_swami_Vivekananda final
 
 <Gallery />
+## Inference
 
+import torch
+from diffusers import FluxPipeline
+
+pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+pipe.load_lora_weights("bekushal/Flux_finetune_swami_vevikananda", weight_name="Flux_finetune_swami_vevikananda.safetensors")
+pipe.fuse_lora(lora_scale=1.0)
+pipe.to("cuda")
+
+prompt = "A young Indian monk with sunglasses in a cricket ground, holding a bat in his hand."
+
+image = pipe(prompt, 
+             num_inference_steps=24, 
+             guidance_scale=3.5,
+             width=512, height=512,
+            ).images[0]
+image.save(f"example.png")
 
 ## Model description