environment upadate, typo correct

Author: DEEP-lianchen

annotator/hed/__init__.py

@@ -90,7 +90,7 @@ def __init__(self):
         self.netNetwork = ControlNetHED_Apache2().float().cuda().eval()
         self.netNetwork.load_state_dict(torch.load(modelpath))
 
-    def __call__(self, input_image, safe=False):
+    def __call__(self, input_image, safe=False, threshold=200):
         assert input_image.ndim == 3
         H, W, C = input_image.shape
         with torch.no_grad():
@@ -106,7 +106,7 @@ def __call__(self, input_image, safe=False):
             edge = (edge * 255.0).clip(0, 255).astype(np.uint8)
 
             content_image = edge
-            content_image[content_image > 200] = 255
+            content_image[content_image > threshold] = 255
             content_image[content_image < 255] = 0
             kernel = np.ones((3,3), np.uint8)
 

assets/image_driven.png

@@ -8,4 +8,5 @@ accelerate==0.21.0
 opencv-python==4.8.1.78
 einops==0.7.0
 botocore==1.34.68
+gradio==3.41.2
 basicsr

assets/text_driven.png

@@ -28,7 +28,7 @@ def main(args):
         raise ValueError("Invalid preprocessor")
 
     if not os.path.isdir(styleshot_model_path):
-        styleshot_lineart_model_path = snapshot_download(styleshot_model_path, local_dir=styleshot_model_path)
+        styleshot_model_path = snapshot_download(styleshot_model_path, local_dir=styleshot_model_path)
         print(f"Downloaded model to {styleshot_model_path}")
 
     # weights for ip-adapter and our content-fusion encoder