Update app.py

app.py

@@ -32,14 +32,13 @@ def run_example(task_prompt, image):
     )
     return parsed_answer
 
-def process_image(image):
+def process_image(image, task_prompt):
     image = Image.fromarray(image)  # Convert NumPy array to PIL Image
     if task_prompt == 'Product Caption':
         task_prompt = '<PC>'
-        results = run_example(task_prompt, image, model_id=model_id)
     elif task_prompt == 'OCR':
         task_prompt = '<OCR>'
-        results = run_example(task_prompt, image, model_id=model_id)
+    
     results = run_example(task_prompt, image)
     
     # Remove the key and get the text value
@@ -76,9 +75,11 @@ document.querySelector('button').addEventListener('click', function() {
     setTimeout(adjustHeight, 500);  // Adjust the height after a small delay to ensure content is loaded
 });
 """
+
 single_task_list =[
     'Product Caption', 'OCR'
 ]
+
 with gr.Blocks(css=css) as demo:
     gr.Markdown(DESCRIPTION)
     with gr.Tab(label="Product Image Select"):
@@ -107,7 +108,7 @@ with gr.Blocks(css=css) as demo:
         ```
         """)
         
-        submit_btn.click(process_image, [input_img], [output_text])
+        submit_btn.click(process_image, [input_img, task_prompt], [output_text])
 
     demo.load(lambda: None, inputs=None, outputs=None, js=js)