Spaces:
Paused
Paused
Update modules/masking_module.py
Browse files
modules/masking_module.py
CHANGED
@@ -137,7 +137,7 @@ def masking_process(image,obj):
|
|
137 |
# task_prompt = '<REGION_TO_SEGMENTATION>'
|
138 |
# # task_prompt = '<OPEN_VOCABULARY_DETECTION>'
|
139 |
# print(type(task_prompt),type(obj))
|
140 |
-
image = image.convert("RGB")
|
141 |
|
142 |
# results = florence2(image,task_prompt, text_input=obj)
|
143 |
# output_image = copy.deepcopy(image)
|
|
|
137 |
# task_prompt = '<REGION_TO_SEGMENTATION>'
|
138 |
# # task_prompt = '<OPEN_VOCABULARY_DETECTION>'
|
139 |
# print(type(task_prompt),type(obj))
|
140 |
+
image = Image.fromarray(image).convert("RGB")
|
141 |
|
142 |
# results = florence2(image,task_prompt, text_input=obj)
|
143 |
# output_image = copy.deepcopy(image)
|