J-LAB commited on
Commit
628b60d
·
verified ·
1 Parent(s): 45845d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -120,7 +120,7 @@ def draw_ocr_bboxes(image, prediction):
120
  def process_image(image, task_prompt, text_input=None, model_id='J-LAB/Florence_2_B_FluxiAI_Product_Caption'):
121
  image = Image.fromarray(image) # Convert NumPy array to PIL Image
122
  if task_prompt == 'Product Caption':
123
- task_prompt = '<MORE_DETAILED_CAPTION>'
124
  results = run_example(task_prompt, image, model_id=model_id)
125
  return results, None
126
  elif task_prompt == 'More Detailed Caption':
@@ -140,7 +140,7 @@ css = """
140
 
141
 
142
  single_task_list =[
143
- 'More Detailed Caption', 'Product Caption'
144
  ]
145
 
146
 
 
120
  def process_image(image, task_prompt, text_input=None, model_id='J-LAB/Florence_2_B_FluxiAI_Product_Caption'):
121
  image = Image.fromarray(image) # Convert NumPy array to PIL Image
122
  if task_prompt == 'Product Caption':
123
+ task_prompt = '<PC>'
124
  results = run_example(task_prompt, image, model_id=model_id)
125
  return results, None
126
  elif task_prompt == 'More Detailed Caption':
 
140
 
141
 
142
  single_task_list =[
143
+ 'Product Caption', 'More Detailed Caption'
144
  ]
145
 
146