add pictures

app.py

@@ -31,6 +31,9 @@ def main():
     st.markdown("1. Split an image into a grid of sub-image patches")  
     st.markdown("2. Embed each patch with a linear projection")  
     st.markdown("3. Each embedded patch becomes a token, and the resulting sequence of embedded patches is the sequence you pass to the model.")  
+    vit=Image.open("pic/vit-figure.jpg")
+    st.image(vit)
+
     st.markdown("I trained the model with 10 epochs, and I use Adam as the optimizer. The accuracy on the test set is 95%.")  
     st.markdown("## Huggingface Space")  
     st.markdown("The page here.")