Update README.md

README.md

@@ -9,6 +9,10 @@ tags:
 - diffusers-training
 base_model: stabilityai/stable-diffusion-xl-base-1.0
 inference: true
+datasets:
+- VikramSingh178/Products-10k-BLIP-captions
+language:
+- en
 ---
 
 <!-- This model card has been generated automatically according to the information the training script had access to. You
@@ -16,18 +20,21 @@ should probably proofread and complete it, then remove this comment. -->
 
 SDXL Product Images
 
-## Intended uses & limitations
+
 
 #### How to use
 
 ```python
-# TODO: add an example code snippet for running this diffusion pipeline
-```
+from diffusers import DiffusionPipeline
+import torch
 
-#### Limitations and bias
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16)
+pipe.to("cuda")
+pipe.load_lora_weights("VikramSingh178/sdxl-lora-finetune-products10k-caption")
 
-[TODO: provide examples of latent issues and potential remediations]
+prompt = "Commercial photography of a black down jacket with a logo on the chest."
 
-## Training details
+image = pipe(prompt, num_inference_steps=45, guidance_scale=7.5).images[0]
+image.save("example.png")
 
-[TODO: describe the data used to train the model]
+```