Update app.py

app.py

@@ -18,10 +18,12 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
 
 models = {
     'J-LAB/Florence-Idesire': AutoModelForCausalLM.from_pretrained('J-LAB/Florence-Idesire', trust_remote_code=True).to("cuda").eval()
+    'J-LAB/Florence_2_L_FluxiAI_Product_Caption': AutoModelForCausalLM.from_pretrained('J-LAB/Florence_2_L_FluxiAI_Product_Caption', trust_remote_code=True).to("cuda").eval()
 }
 
 processors = {
     'J-LAB/Florence-Idesire': AutoProcessor.from_pretrained('J-LAB/Florence-Idesire', trust_remote_code=True),
+    'J-LAB/Florence_2_L_FluxiAI_Product_Caption': AutoProcessor.from_pretrained('J-LAB/Florence_2_L_FluxiAI_Product_Caption', trust_remote_code=True),
 }
 
 
@@ -117,8 +119,8 @@ def draw_ocr_bboxes(image, prediction):
 
 def process_image(image, task_prompt, text_input=None, model_id='J-LAB/Florence-Idesire'):
     image = Image.fromarray(image)  # Convert NumPy array to PIL Image
-    if task_prompt == 'Caption':
-        task_prompt = '<CAPTION>'
+    if task_prompt == 'More Detailed Caption':
+        task_prompt = '<MORE_DETAILED_CAPTION>'
         results = run_example(task_prompt, image, model_id=model_id)
         return results, None
     elif task_prompt == 'Detailed Caption':