Update README.md

vantunes-crl · Nov 23, 2022 · 765dc7c · 765dc7c
1 parent c0b8efc
commit 765dc7c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -14,12 +14,12 @@ images = model.generate_text2img('кошка', batch_size=4, h=512, w=512, num_s
 ```
 ## 2. img2img
 ```python
-from natalle.natalle_model import Natalle
+from kandinsky2 import get_kandinsky2
 from PIL import Image
 
-model = Natalle('inference.yaml', 'natalle.pt', 'cuda', task_type='img2img')
+model = get_kandinsky2('cuda', task_type='text2img')
 init_image = Image.open('image.jpg')
-images = model.generate_img2img('a photo of realistic landscape', init_image, strength=0.6)
+images = model.generate_img2img('кошка',  init_image, strength=0.8, num_steps=50, denoised_type='dynamic_threshold', dynamic_threshold_v=99.5, sampler='ddim_sampler', ddim_eta=0.05, guidance_scale=10)
 ```
 ## 3. inpainting
 ```python