Spaces:
Sleeping
Sleeping
Update depth_app.py
Browse files- depth_app.py +2 -2
depth_app.py
CHANGED
@@ -65,10 +65,10 @@ def main():
|
|
65 |
st.title("Monodepth2 Depth Estimation")
|
66 |
st.write("Upload a PNG image to get depth estimation")
|
67 |
|
68 |
-
if not os.path.
|
69 |
subprocess.run(["wget", "https://github.com/meghakalia/depthEstimationColonoscopy/releases/download/0.0.1/depth.pth"])
|
70 |
|
71 |
-
if not os.path.
|
72 |
subprocess.run(["wget", "https://github.com/meghakalia/depthEstimationColonoscopy/releases/download/0.0.1/encoder.pth"])
|
73 |
|
74 |
uploaded_file = st.file_uploader("Choose a PNG file", type="png")
|
|
|
65 |
st.title("Monodepth2 Depth Estimation")
|
66 |
st.write("Upload a PNG image to get depth estimation")
|
67 |
|
68 |
+
if not os.path.exists("depth.pth"):
|
69 |
subprocess.run(["wget", "https://github.com/meghakalia/depthEstimationColonoscopy/releases/download/0.0.1/depth.pth"])
|
70 |
|
71 |
+
if not os.path.exists("encoder.pth"):
|
72 |
subprocess.run(["wget", "https://github.com/meghakalia/depthEstimationColonoscopy/releases/download/0.0.1/encoder.pth"])
|
73 |
|
74 |
uploaded_file = st.file_uploader("Choose a PNG file", type="png")
|