Update README.md

README.md

@@ -18,12 +18,15 @@ Here we give an example model fine-tuned using 5 images of a cat downloaded from
 ```python
 
 from diffusers import StableDiffusionPipeline
+from src import diffuser_training
 
 device = 'cuda'
-model_id = 'nupurkmr9/custom_diffusion_cat/cat'
+model_id = "CompVis/stable-diffusion-v1-4"
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to(device)
 
+diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, 'custom_diffusion_cat.bin')
+
 prompt = "<new1> cat swimming in a pool"
 images = pipe(prompt, num_inference_steps=200, guidance_scale=6., eta=1.).images
 ```