z-uo commited on
Commit
c528095
1 Parent(s): 680e883

put also show under the spinner

Browse files
Files changed (1) hide show
  1. app.py +16 -16
app.py CHANGED
@@ -185,19 +185,19 @@ else:
185
 
186
  with st.spinner("Waiting for the predictions..."):
187
  pil_scaled, pil_depth, html_string = predict(pil_image)
188
- components.html(html_string)
189
- #st.markdown(html_string, unsafe_allow_html=True)
190
-
191
- col1, col2, col3 = st.columns(3)
192
- with col1:
193
- st.image(pil_scaled)
194
- with col2:
195
- st.image(pil_depth)
196
- with col3:
197
- with open('model.obj') as f:
198
- st.download_button('Download model.obj', f, file_name="model.obj")
199
- os.remove('model.obj')
200
- pil_depth.save('tmp.png')
201
- with open('tmp.png', "rb") as f:
202
- st.download_button('Download depth.png', f,file_name="depth.png", mime="image/png")
203
- os.remove('tmp.png')
 
185
 
186
  with st.spinner("Waiting for the predictions..."):
187
  pil_scaled, pil_depth, html_string = predict(pil_image)
188
+ components.html(html_string)
189
+ #st.markdown(html_string, unsafe_allow_html=True)
190
+
191
+ col1, col2, col3 = st.columns(3)
192
+ with col1:
193
+ st.image(pil_scaled)
194
+ with col2:
195
+ st.image(pil_depth)
196
+ with col3:
197
+ with open('model.obj') as f:
198
+ st.download_button('Download model.obj', f, file_name="model.obj")
199
+ os.remove('model.obj')
200
+ pil_depth.save('tmp.png')
201
+ with open('tmp.png', "rb") as f:
202
+ st.download_button('Download depth.png', f,file_name="depth.png", mime="image/png")
203
+ os.remove('tmp.png')