Beasto commited on
Commit
3c66367
·
verified ·
1 Parent(s): 6cd8731

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -0
src/streamlit_app.py CHANGED
@@ -265,6 +265,7 @@ user_input = st.text_input("Enter some text:", "")
265
 
266
  if st.button("Generate Image"):
267
  with st.spinner("Generating image..."):
 
268
  output_tokens = generate_image_tokens(user_input)
269
  embedding = get_embeddings(output_tokens, codebook)
270
  image = decoder(embedding)[0].numpy()
 
265
 
266
  if st.button("Generate Image"):
267
  with st.spinner("Generating image..."):
268
+ st.write(user_input)
269
  output_tokens = generate_image_tokens(user_input)
270
  embedding = get_embeddings(output_tokens, codebook)
271
  image = decoder(embedding)[0].numpy()