DamarJati commited on
Commit
7fa3baf
β€’
1 Parent(s): ced21ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ def process_images_florence_lama(image, model_choice):
73
  image_cv = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
74
 
75
  # Run Florence to get mask
76
- text_input = 'watermark, text' # Teks untuk Florence agar mengenali watermark
77
  task_prompt = '<REGION_TO_SEGMENTATION>'
78
  image_pil = Image.fromarray(image_cv) # Convert array to PIL Image
79
  inputs = florence_processor(text=task_prompt + text_input, images=image_pil, return_tensors="pt").to("cuda")
 
73
  image_cv = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
74
 
75
  # Run Florence to get mask
76
+ text_input = 'watermark'
77
  task_prompt = '<REGION_TO_SEGMENTATION>'
78
  image_pil = Image.fromarray(image_cv) # Convert array to PIL Image
79
  inputs = florence_processor(text=task_prompt + text_input, images=image_pil, return_tensors="pt").to("cuda")