Update app.py

app.py

@@ -50,7 +50,7 @@ model, processor, device = initialize_model()
 #     generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
 #     return processor.post_process_generation(generated_text, task=task_prompt, image_size=(image.size[0], image.size[1]))
 
-def run_example(task_prompt, text_input=None):
+def run_example(task_prompt, image, text_input=None):
     if text_input is None:
         prompt = task_prompt
     else: