Update inference.py

inference.py

@@ -39,7 +39,7 @@ def process_inference_results(results, process_image=False):
 
     return extracted_texts
 
-def inference_and_run(image_path, prompt, conv_mode="ferret_gemma_instruct", model_path="jadechoghari/Ferret-UI-Gemma2b", box=None):
+def inference_and_run(image_path, prompt, conv_mode="ferret_gemma_instruct", model_path="jadechoghari/Ferret-UI-Gemma2b", box=None, process_image=False):
     """
     Run the inference and capture the errors for debugging.
     """
@@ -87,7 +87,7 @@ def inference_and_run(image_path, prompt, conv_mode="ferret_gemma_instruct", mod
                 with open(output_file_path, "r") as output_file:
                     results = [json.loads(line) for line in output_file]
                 
-                return process_inference_results(results)
+                return process_inference_results(results, process_image)
             else:
                 print("No output JSONL files found.")
                 return None, None