Spaces:
Build error
Build error
jhj0517 commited on
Commit ·
607e627
1
Parent(s): fbd23d6
rollback pytoshop because it doesn't work in huggingface
Browse files- modules/html_constants.py +2 -1
- modules/mask_utils.py +1 -2
- modules/sam.py +1 -2
- requirements.txt +1 -1
modules/html_constants.py
CHANGED
|
@@ -14,7 +14,8 @@ PROJECT_NAME = """
|
|
| 14 |
"""
|
| 15 |
|
| 16 |
MARKDOWN_NOTE = """
|
| 17 |
-
**If you want to run your own local GPU , please visit this [repository](https://github.com/jhj0517/Layer-Divider-WebUI).
|
|
|
|
| 18 |
"""
|
| 19 |
|
| 20 |
PARAMS_EXPLANATION = """
|
|
|
|
| 14 |
"""
|
| 15 |
|
| 16 |
MARKDOWN_NOTE = """
|
| 17 |
+
**If you want to run your own local GPU , please visit this [repository](https://github.com/jhj0517/Layer-Divider-WebUI).
|
| 18 |
+
The repository also supports layered PSD (Photoshop) file!**
|
| 19 |
"""
|
| 20 |
|
| 21 |
PARAMS_EXPLANATION = """
|
modules/mask_utils.py
CHANGED
|
@@ -87,7 +87,6 @@ def save_psd(input_image_data, layer_data, layer_names, blending_modes):
|
|
| 87 |
timestamp = datetime.now().strftime("%m%d%H%M%S")
|
| 88 |
with open(f"outputs/psd/result-{timestamp}.psd", 'wb') as output_file:
|
| 89 |
psd_file.write(output_file)
|
| 90 |
-
return f"outputs/psd/result-{timestamp}.psd"
|
| 91 |
|
| 92 |
|
| 93 |
def save_psd_with_masks(image, masks):
|
|
@@ -95,6 +94,6 @@ def save_psd_with_masks(image, masks):
|
|
| 95 |
mask_layers = create_mask_layers(image, masks)
|
| 96 |
names = [f'Part {i}' for i in range(len(mask_layers))]
|
| 97 |
modes = [BlendMode.normal] * (len(mask_layers)+1)
|
| 98 |
-
|
| 99 |
|
| 100 |
|
|
|
|
| 87 |
timestamp = datetime.now().strftime("%m%d%H%M%S")
|
| 88 |
with open(f"outputs/psd/result-{timestamp}.psd", 'wb') as output_file:
|
| 89 |
psd_file.write(output_file)
|
|
|
|
| 90 |
|
| 91 |
|
| 92 |
def save_psd_with_masks(image, masks):
|
|
|
|
| 94 |
mask_layers = create_mask_layers(image, masks)
|
| 95 |
names = [f'Part {i}' for i in range(len(mask_layers))]
|
| 96 |
modes = [BlendMode.normal] * (len(mask_layers)+1)
|
| 97 |
+
save_psd(image, original_layer+mask_layers, ['Original_Image']+names, modes)
|
| 98 |
|
| 99 |
|
modules/sam.py
CHANGED
|
@@ -62,7 +62,6 @@ class SamInference:
|
|
| 62 |
masks = self.mask_generator.generate(image)
|
| 63 |
combined_image = create_mask_combined_images(image, masks)
|
| 64 |
gallery = create_mask_gallery(image, masks)
|
| 65 |
-
|
| 66 |
-
return [combined_image] + gallery, psd
|
| 67 |
except Exception as e:
|
| 68 |
print(e)
|
|
|
|
| 62 |
masks = self.mask_generator.generate(image)
|
| 63 |
combined_image = create_mask_combined_images(image, masks)
|
| 64 |
gallery = create_mask_gallery(image, masks)
|
| 65 |
+
return [combined_image] + gallery
|
|
|
|
| 66 |
except Exception as e:
|
| 67 |
print(e)
|
requirements.txt
CHANGED
|
@@ -7,4 +7,4 @@ matplotlib
|
|
| 7 |
onnxruntime
|
| 8 |
onnx
|
| 9 |
gradio
|
| 10 |
-
pytoshop
|
|
|
|
| 7 |
onnxruntime
|
| 8 |
onnx
|
| 9 |
gradio
|
| 10 |
+
pytoshop==1.2.0
|