Update app.py

app.py

@@ -1,3 +1,15 @@
+
+import os
+from huggingface_hub import login
+from huggingface_hub import Repository
+from huggingface_hub import hf_hub_download
+# os.system("python -m pip install --upgrade pip")
+os.system("pip uninstall -y gradio")
+os.system("pip install gradio==3.27.0")
+
+token = os.environ['HUB_TOKEN']
+hf_hub_download(repo_id="JunchuanYu/files_for_segmentRS", filename="utils.py",local_dir='.',token=token)
+
 import os
 import cv2
 import matplotlib
@@ -11,20 +23,9 @@ from PIL import Image
 from segment_anything import SamAutomaticMaskGenerator, SamPredictor, sam_model_registry
 import logging
 from utils import *
-from huggingface_hub import login
-from huggingface_hub import Repository
-# os.system("python -m pip install --upgrade pip")
-os.system("pip uninstall -y gradio")
-os.system("pip install gradio==3.27.0")
 
-login(token = os.environ['HUB_TOKEN'])
-repo = Repository(
-    local_dir="files",
-    repo_type="dataset",
-    clone_from="JunchuanYu/files_for_segmentRS",
-    token=True
-)
-repo.git_pull()
+
+
 matplotlib.pyplot.switch_backend('Agg') # for matplotlib to work in gradio
 #setup model
 sam_checkpoint = "sam_vit_h_4b8939.pth"