patrawtf commited on
Commit
11b967a
·
1 Parent(s): 5becf68

generate button in the middle

Browse files
Files changed (1) hide show
  1. app.py +11 -10
app.py CHANGED
@@ -68,16 +68,17 @@ def main():
68
  with col2:
69
  factor_slider = st.slider('Choose Upscaling Factor:', min_value=1.0, max_value=10.0, value=1.0, step=0.5, format="")
70
  st.write(f"Selected Upscaling Factor: {factor_slider}")
71
-
72
- if st.button("Generate"):
73
- factor = st.session_state.user_factor if st.session_state.user_factor is not None else factor_slider
74
- with st.spinner('Generating images...'):
75
- result_image_stream = asyncio.run(async_upscaler(Image_stream, factor))
76
- try:
77
- print(type(result_image_stream))
78
- st.image(result_image_stream.getvalue())
79
- except Exception as e:
80
- st.error(f"An error occurred during image upscaling: {e}")
 
81
 
82
 
83
  if __name__ == "__main__":
 
68
  with col2:
69
  factor_slider = st.slider('Choose Upscaling Factor:', min_value=1.0, max_value=10.0, value=1.0, step=0.5, format="")
70
  st.write(f"Selected Upscaling Factor: {factor_slider}")
71
+ gen_col1, gen_col2, gen_col3 = st.columns([3, 1, 3])
72
+ with gen_col2:
73
+ if st.button("Generate"):
74
+ factor = st.session_state.user_factor if st.session_state.user_factor is not None else factor_slider
75
+ with st.spinner('Generating images...'):
76
+ result_image_stream = asyncio.run(async_upscaler(Image_stream, factor))
77
+ try:
78
+ print(type(result_image_stream))
79
+ st.image(result_image_stream.getvalue())
80
+ except Exception as e:
81
+ st.error(f"An error occurred during image upscaling: {e}")
82
 
83
 
84
  if __name__ == "__main__":