Dagfinn1962 commited on
Commit
f28b6fb
1 Parent(s): bab0afc

Update streamlit_app.py

Browse files
Files changed (1) hide show
  1. streamlit_app.py +2 -2
streamlit_app.py CHANGED
@@ -103,7 +103,7 @@ def query_vispunk(prompt):
103
 
104
 
105
  def main(prompt):
106
- show_app.write("**You:** " + prompt)
107
  if st.session_state.image_choice:
108
  image = huggingface_text_to_image(prompt)
109
  else:
@@ -114,7 +114,7 @@ def main(prompt):
114
  with st.sidebar:
115
  st.session_state.image_choice = st.toggle(st.session_state.image_choice_name,value=st.session_state.image_choice,on_change=image_choice)
116
  if st.session_state.image_choice:
117
- st.session_state.draw_model = st.selectbox('Draw Models', sorted(st.session_state.draw_model_list.keys(),key=lambda x:x.split("-")[0]),on_change=change_paramater)
118
  else:
119
  pass
120
 
 
103
 
104
 
105
  def main(prompt):
106
+ show_app.write("**Your Prompt:** " + prompt)
107
  if st.session_state.image_choice:
108
  image = huggingface_text_to_image(prompt)
109
  else:
 
114
  with st.sidebar:
115
  st.session_state.image_choice = st.toggle(st.session_state.image_choice_name,value=st.session_state.image_choice,on_change=image_choice)
116
  if st.session_state.image_choice:
117
+ st.session_state.draw_model = st.selectbox('SafeSensors/Models', sorted(st.session_state.draw_model_list.keys(),key=lambda x:x.split("-")[0]),on_change=change_paramater)
118
  else:
119
  pass
120