Update inference.py

inference.py

@@ -6,7 +6,7 @@ import re
 import json
 import subprocess
 
-def process_inference_results(results):
+def process_inference_results(results, process_image=False):
     """
     Process the inference results by:
     1. Adding bounding boxes on the image based on the coordinates in 'text'.
@@ -34,7 +34,10 @@ def process_inference_results(results):
 
         processed_images.append(img)
 
-    return processed_images, extracted_texts
+    if process_image:
+        return processed_images, extracted_texts
+
+    return extracted_texts
 
 def inference_and_run(image_path, prompt, conv_mode="ferret_gemma_instruct", model_path="jadechoghari/Ferret-UI-Gemma2b", box=None):
     """