diff --git a/README.md b/README.md
index 904288b..181f51a 100644
--- a/README.md
+++ b/README.md
@@ -67,7 +67,7 @@ streamlit run generate.py -- --model sanguogpt.pth -l 100 --webui
 
 Here is an example of the UI:
 
-<img src="images/generate-v0.1.png" alt="detailed model architecture" width="500"/>
+<img src="images/generate-v0.2.png" alt="detailed model architecture" width="500"/>
 
 ## Dataset
 
diff --git a/images/generate-v0.2.png b/images/generate-v0.2.png
new file mode 100644
index 0000000..ead3b26
Binary files /dev/null and b/images/generate-v0.2.png differ