Spaces:
Running
Running
put also show under the spinner
Browse files
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 |
-
|
194 |
-
with col2:
|
195 |
-
|
196 |
-
with col3:
|
197 |
-
|
198 |
-
|
199 |
-
|
200 |
-
|
201 |
-
|
202 |
-
|
203 |
-
|
|
|
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')
|