Update README.md

README.md

@@ -16,21 +16,14 @@ Here we give an example model fine-tuned using 5 images of a cat downloaded from
 
 ## Example code of inference
 ```python
-import os
-import sys
-import torch
 
-os.system("git clone https://github.com/adobe-research/custom-diffusion")
-sys.path.append("custom-diffusion")
 from diffusers import StableDiffusionPipeline
-from src import diffuser_training
 
 device = 'cuda'
+model_id = 'nupurkmr9/custom_diffusion_cat/cat'
 pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe = pipe.to(device)
 
-weight_path = 'custom_diffusion_cat.bin'
-diffuser_training.load_model(pipe.text_encoder, pipe.tokenizer, pipe.unet, weight_path, '<new1>')
 prompt = "<new1> cat swimming in a pool"
 images = pipe(prompt, num_inference_steps=200, guidance_scale=6., eta=1.).images
 ```