Update depth_app.py

depth_app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-import subprocess
 import numpy as np
 import PIL.Image as pil
 import torch
@@ -11,6 +10,26 @@ import matplotlib.cm as cm
 
 from resnet_encoder import ResnetEncoder
 from depth_decoder import DepthDecoder
+
+import os
+import subprocess
+import requests
+
+def download_model_files():
+    depth_url = "https://github.com/meghakalia/depthEstimationColonoscopy/releases/download/0.0.1/depth.pth"
+    encoder_url = "https://github.com/meghakalia/depthEstimationColonoscopy/releases/download/0.0.1/encoder.pth"
+
+    if not os.path.exists("depth.pth"):
+        response = requests.get(depth_url)
+        with open("depth.pth", "wb") as f:
+            f.write(response.content)
+        
+    if not os.path.exists("encoder.pth"):
+        response = requests.get(encoder_url)
+        with open("encoder.pth", "wb") as f:
+            f.write(response.content)
+
+
 # from pose_decoder import PoseDecoder
 # from pose_cnn import PoseCNN
 
@@ -66,12 +85,8 @@ def main():
     st.title("Monodepth2 Depth Estimation")
     st.write("Upload a PNG image to get depth estimation")
 
-    if not os.path.exists("depth.pth"):
-        subprocess.run(["wget", "https://github.com/meghakalia/depthEstimationColonoscopy/releases/download/0.0.1/depth.pth"])
-        
-    if not os.path.exists("encoder.pth"):
-        subprocess.run(["wget", "https://github.com/meghakalia/depthEstimationColonoscopy/releases/download/0.0.1/encoder.pth"])
-                
+    download_model_files()
+    
     uploaded_file = st.file_uploader("Choose a PNG file", type="png")
     if uploaded_file is not None:
         image = pil.open(uploaded_file).convert('RGB')