Update src/streamlit_app.py

src/streamlit_app.py

@@ -251,4 +251,4 @@ if st.button("Generate Image"):
         embedding = get_embeddings(output_tokens, codebook)
         image = decoder(embedding)[0].numpy()
         image = np.clip(image * 255, 0, 255).astype(np.uint8)
-        st.image(image)
+        st.image(image, caption="Generated Image", width=512)