Replace shell commands with Python code

app.py

@@ -1,21 +1,33 @@
 
+import os
+import subprocess
+import requests
+import zipfile
 import gradio as gr
 import torch
 import numpy as np
 from torchvision.transforms import ToTensor
 from PIL import Image
 import cv2
-import zipfile
 
 # Ensure the necessary model files are available
-!wget -q https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth
-!mkdir -p weights
-!mv sam_vit_h_4b8939.pth weights/
+def download_file(url, destination):
+    response = requests.get(url, stream=True)
+    with open(destination, 'wb') as f:
+        f.write(response.content)
+
+# Download SAM model
+if not os.path.exists("weights/sam_vit_h_4b8939.pth"):
+    os.makedirs("weights", exist_ok=True)
+    download_file("https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth", "weights/sam_vit_h_4b8939.pth")
+
+# Clone EfficientSAM repository if not already cloned
+if not os.path.exists("EfficientSAM"):
+    subprocess.run(["git", "clone", "https://github.com/yformer/EfficientSAM.git"])
+
+# Install dependencies
+subprocess.run(["pip", "install", "git+https://github.com/facebookresearch/segment-anything.git"])
 
-!git clone https://github.com/yformer/EfficientSAM.git
-import os
-os.chdir("EfficientSAM")
-!pip install git+https://github.com/facebookresearch/segment-anything.git
 from segment_anything import sam_model_registry, SamAutomaticMaskGenerator
 from efficient_sam.build_efficient_sam import build_efficient_sam_vits
 
@@ -156,3 +168,4 @@ interface = gr.Interface(
 )
 
 interface.launch(debug=True)
+