bug fix

app.py

@@ -65,6 +65,12 @@ model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
 
 @spaces.GPU
 def run_grounding(input_image, grounding_caption, box_threshold, text_threshold):
+    # Convert numpy array to PIL Image if needed
+    if isinstance(input_image, np.ndarray):
+        if input_image.ndim == 3:
+            input_image = cv2.cvtColor(input_image, cv2.COLOR_BGR2RGB)
+        input_image = Image.fromarray(input_image)
+    
     init_image = input_image.convert("RGB")
     original_size = init_image.size
 
@@ -100,7 +106,7 @@ if __name__ == "__main__":
 
         with gr.Row():
             with gr.Column():
-                input_image = gr.Image(label="Input Image")
+                input_image = gr.Image(label="Input Image", type="pil")
                 grounding_caption = gr.Textbox(label="Detection Prompt")
                 run_button = gr.Button("Run")