m

handler.py

@@ -4,52 +4,17 @@ import torch
 import requests
 from PIL import Image
 from io import BytesIO
-
-just_get_sd_mask_function = None
+from test import just_get_sd_mask
 
 print(os.listdir('/usr/local/'))
 print(torch.version.cuda)
 
 class EndpointHandler():
     def __init__(self, path="."):
-        global just_get_sd_mask_function
-        
-        is_production = True
-        
-        if False:
-            return
-
-        os.chdir(path)
-        
-        os.environ['AM_I_DOCKER'] = 'False'
-        os.environ['BUILD_WITH_CUDA'] = 'True'
-        os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/' if is_production else '/usr/local/cuda-12.1/'
-
-        # Install Segment Anything
-        subprocess.run(["python", "-m", "pip", "install", "-e", "segment_anything"])
-
-        # Install Grounding DINO
-        subprocess.run(["python", "-m", "pip", "install", "-e", "GroundingDINO"])
-    
-        subprocess.run("wget https://huggingface.co/Uminosachi/sam-hq/resolve/main/sam_hq_vit_h.pth -O ./sam_hq_vit_h.pth", shell=True)
-
-        # Install diffusers
-        subprocess.run(["pip", "install", "--upgrade", "diffusers[torch]"])
-
-        # Install osx
-        subprocess.run(["git", "submodule", "update", "--init", "--recursive"])
-        subprocess.run(["bash", "grounded-sam-osx/install.sh"], cwd="grounded-sam-osx")
-
-        # Install RAM & Tag2Text
-        subprocess.run(["git", "clone", "https://github.com/xinyu1205/recognize-anything.git"])
-        subprocess.run(["pip", "install", "-r", "./recognize-anything/requirements.txt"])
-        subprocess.run(["pip", "install", "-e", "./recognize-anything/"])
-
-        from test import just_get_sd_mask
-        just_get_sd_mask_function = just_get_sd_mask
+        pass
 
     def __call__(self, data):
-        mask_pil = just_get_sd_mask_function(Image.open("assets/demo1.jpg"), "bear", 10)
+        mask_pil = just_get_sd_mask(Image.open("assets/demo1.jpg"), "bear", 10)
 
         if mask_pil.mode != 'RGB':
             mask_pil = mask_pil.convert('RGB')

test.py

@@ -1,3 +1,37 @@
+import os
+import subprocess
+import torch
+import requests
+from PIL import Image
+from io import BytesIO
+
+is_production = True
+
+os.chdir("/repository" if is_production else ".")
+os.environ['AM_I_DOCKER'] = 'False'
+os.environ['BUILD_WITH_CUDA'] = 'True'
+os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/' if is_production else '/usr/local/cuda-12.1/'
+
+# Install Segment Anything
+subprocess.run(["python", "-m", "pip", "install", "-e", "segment_anything"])
+
+# Install Grounding DINO
+subprocess.run(["python", "-m", "pip", "install", "-e", "GroundingDINO"])
+
+subprocess.run("wget https://huggingface.co/Uminosachi/sam-hq/resolve/main/sam_hq_vit_h.pth -O ./sam_hq_vit_h.pth", shell=True)
+
+# Install diffusers
+subprocess.run(["pip", "install", "--upgrade", "diffusers[torch]"])
+
+# Install osx
+subprocess.run(["git", "submodule", "update", "--init", "--recursive"])
+subprocess.run(["bash", "grounded-sam-osx/install.sh"], cwd="grounded-sam-osx")
+
+# Install RAM & Tag2Text
+subprocess.run(["git", "clone", "https://github.com/xinyu1205/recognize-anything.git"])
+subprocess.run(["pip", "install", "-r", "./recognize-anything/requirements.txt"])
+subprocess.run(["pip", "install", "-e", "./recognize-anything/"])
+
 from grounded_sam_demo import grounded_sam_demo
 import numpy as np
 from PIL import Image