default image

app.py

@@ -180,21 +180,23 @@ uploader = st.file_uploader('Upload your portrait here',type=['jpg','jpeg','png'
 
 if uploader is not None:
     pil_image = Image.open(uploader)
-    pil_scaled, pil_depth, html_string = predict(pil_image)
-
-    components.html(html_string)
-    #st.markdown(html_string, unsafe_allow_html=True)
-
-    col1, col2, col3 = st.columns(3)
-    with col1:
-        st.image(pil_scaled)
-    with col2:
-        st.image(pil_depth)
-    with col3:
-        with open('model.obj') as f:
-            st.download_button('Download model.obj', f, file_name="model.obj")
-        os.remove('model.obj')
-        pil_depth.save('tmp.png')
-        with open('tmp.png', "rb") as f:
-            st.download_button('Download depth.png', f,file_name="depth.png", mime="image/png")
-        os.remove('tmp.png')
+else:
+    pil_image = Image.open('119_image.png')
+pil_scaled, pil_depth, html_string = predict(pil_image)
+
+components.html(html_string)
+#st.markdown(html_string, unsafe_allow_html=True)
+
+col1, col2, col3 = st.columns(3)
+with col1:
+    st.image(pil_scaled)
+with col2:
+    st.image(pil_depth)
+with col3:
+    with open('model.obj') as f:
+        st.download_button('Download model.obj', f, file_name="model.obj")
+    os.remove('model.obj')
+    pil_depth.save('tmp.png')
+    with open('tmp.png', "rb") as f:
+        st.download_button('Download depth.png', f,file_name="depth.png", mime="image/png")
+    os.remove('tmp.png')