Beasto commited on
Commit
85e2b48
·
verified ·
1 Parent(s): 0af326c

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -244,7 +244,7 @@ def get_embeddings(indices, codebook):
244
  transformer.load_weights('src/VQGAN_Transformer.weights.h5')
245
  user_input = st.text_input("Enter some text:", "")
246
 
247
- if st.button("Generate Image"):
248
  with st.spinner("Generating image..."):
249
  st.write(user_input)
250
  output_tokens = generate_image_tokens(user_input)
 
244
  transformer.load_weights('src/VQGAN_Transformer.weights.h5')
245
  user_input = st.text_input("Enter some text:", "")
246
 
247
+ if user_input != "":
248
  with st.spinner("Generating image..."):
249
  st.write(user_input)
250
  output_tokens = generate_image_tokens(user_input)