Update README.md

README.md

@@ -2,6 +2,8 @@
 license: apache-2.0
 ---
 
+More detail in: https://github.com/Qiyuan-Ge/PaintMind
+
 # PaintMind
 
 ## Install
@@ -15,6 +17,19 @@ import paintmind as pm
 ````
 
 ## Reconstruction
+````
+img = Image.open(img_path).convert('RGB')
+img = pm.stage1_transform(is_train=False)(img)
+# load pretrained vit-vqgan
+model = pm.create_model(arch='vqgan', version='vit-s-vqgan', pretrained=True)
+# encode image to latent
+z, _, _ = model.encode(img.unsqueeze(0))
+# decode latent to image
+rec = model.decode(z).squeeze(0)
+rec = torch.clamp(rec, -1., 1.)
+````
+
+
 #### 1.
 ````
 pm.reconstruction(img_url='https://cdn.pixabay.com/photo/2014/10/22/15/47/squirrel-498139_960_720.jpg')
@@ -55,4 +70,6 @@ pm.reconstruction(img_url='https://cdn.pixabay.com/photo/2017/10/28/07/47/woman-
 <img src="https://github.com/Qiyuan-Ge/PaintMind/blob/main/assets/rec_5.png?raw=true">
 </div>
 
-````
+## Text2Image
+Not finish yet~~
+