m

GroundingDINO/groundingdino/datasets/transforms.py

@@ -10,8 +10,8 @@ import torch
 import torchvision.transforms as T
 import torchvision.transforms.functional as F
 
-from groundingdino.util.box_ops import box_xyxy_to_cxcywh
-from groundingdino.util.misc import interpolate
+from .groundingdino.util.box_ops import box_xyxy_to_cxcywh
+from .groundingdino.util.misc import interpolate
 
 
 def crop(image, target, region):

handler.py

@@ -1,16 +1,19 @@
 import os
 import subprocess
 import torch
-from test import just_get_sd_mask
 import requests
 from PIL import Image
 from io import BytesIO
 
+just_get_sd_mask_function = None
+
 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:
@@ -42,6 +45,9 @@ class EndpointHandler():
         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
+
     def __call__(self, data):
         mask_pil = just_get_sd_mask(Image.open("assets/demo1.jpg"), "bear", 10)