Update app.py
Browse files
app.py
CHANGED
|
@@ -77,7 +77,7 @@ def process_images_florence_lama(image, model_choice):
|
|
| 77 |
image_cv = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
|
| 78 |
|
| 79 |
# Run Florence to get mask
|
| 80 |
-
text_input = 'watermark'
|
| 81 |
task_prompt = '<REGION_TO_SEGMENTATION>'
|
| 82 |
image_pil = Image.fromarray(image_cv) # Convert array to PIL Image
|
| 83 |
inputs = florence_processor(text=task_prompt + text_input, images=image_pil, return_tensors="pt").to("cuda")
|
|
|
|
| 77 |
image_cv = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
|
| 78 |
|
| 79 |
# Run Florence to get mask
|
| 80 |
+
text_input = 'watermark, text, logo, company name'
|
| 81 |
task_prompt = '<REGION_TO_SEGMENTATION>'
|
| 82 |
image_pil = Image.fromarray(image_cv) # Convert array to PIL Image
|
| 83 |
inputs = florence_processor(text=task_prompt + text_input, images=image_pil, return_tensors="pt").to("cuda")
|