Try2

app.py

@@ -10,6 +10,7 @@ import gradio as gr
 import os
 import requests
 import zipfile
+import subprocess
 
 # Ensure the necessary model files are available
 def download_file(url, destination):
@@ -17,20 +18,13 @@ def download_file(url, destination):
     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"):
-    os.makedirs("EfficientSAM", exist_ok=True)
-    os.system("git clone https://github.com/yformer/EfficientSAM.git EfficientSAM")
+# Install the necessary packages
+subprocess.run(["pip", "install", "git+https://github.com/facebookresearch/segment-anything.git"])
+subprocess.run(["git", "clone", "https://github.com/yformer/EfficientSAM.git"])
 
+# Change directory to EfficientSAM
 os.chdir("EfficientSAM")
 
-!pip install git+https://github.com/facebookresearch/segment-anything.git
-
 from segment_anything.utils.amg import (
     batched_mask_to_box,
     calculate_stability_score,
@@ -166,3 +160,4 @@ interface = gr.Interface(
 
 interface.launch(debug=True)
 
+