Santhosh Subramanian commited on
Commit
724e585
·
1 Parent(s): e52db89

adding aws limit increase

Browse files
Files changed (1) hide show
  1. app.py +16 -12
app.py CHANGED
@@ -71,9 +71,11 @@ with face_images.form("my_form"):
71
  )
72
  submitted = st.form_submit_button("Submit")
73
  if submitted:
74
- st.session_state["s3_face_file_path"] = zip_and_upload_images(
75
- identifier, uploaded_files, "face"
76
- )
 
 
77
 
78
  preset_theme_images = st.empty()
79
  with preset_theme_images.form("choose-preset-theme"):
@@ -125,15 +127,17 @@ if st.session_state["view"]:
125
  title = st.text_input("Theme Name")
126
  submitted_3 = st.form_submit_button("Submit!")
127
  if submitted_3:
128
- st.session_state["s3_theme_file_path"] = zip_and_upload_images(
129
- identifier, uploaded_files_2, "theme"
130
- )
131
- st.session_state["model_inputs"] = {
132
- "superhero_file_path": st.session_state["s3_theme_file_path"],
133
- "person_file_path": st.session_state["s3_face_file_path"],
134
- "superhero_prompt": title,
135
- "num_images": 50,
136
- }
 
 
137
 
138
  train = st.empty()
139
  with train.form("training"):
 
71
  )
72
  submitted = st.form_submit_button("Submit")
73
  if submitted:
74
+ with st.spinner('Uploading...'):
75
+ st.session_state["s3_face_file_path"] = zip_and_upload_images(
76
+ identifier, uploaded_files, "face"
77
+ )
78
+ st.success('Done!')
79
 
80
  preset_theme_images = st.empty()
81
  with preset_theme_images.form("choose-preset-theme"):
 
127
  title = st.text_input("Theme Name")
128
  submitted_3 = st.form_submit_button("Submit!")
129
  if submitted_3:
130
+ with st.spinner('Uploading...'):
131
+ st.session_state["s3_theme_file_path"] = zip_and_upload_images(
132
+ identifier, uploaded_files_2, "theme"
133
+ )
134
+ st.session_state["model_inputs"] = {
135
+ "superhero_file_path": st.session_state["s3_theme_file_path"],
136
+ "person_file_path": st.session_state["s3_face_file_path"],
137
+ "superhero_prompt": title,
138
+ "num_images": 50,
139
+ }
140
+ st.success('Done!')
141
 
142
  train = st.empty()
143
  with train.form("training"):