Update app.py

app.py

@@ -23,7 +23,7 @@ def preprocess_image(image_path, max_size=(800, 800)):
         image.thumbnail(max_size, Image.LANCZOS)
 
     # Convert image to numpy array
-    image_np = np.array(image)
+    image_np = np.array(image, dtype=np.float32)  # Ensure the array is float32
 
     # Ensure the image is in the format [height, width, channels]
     if image_np.ndim == 2:  # Grayscale image
@@ -65,7 +65,7 @@ def run_florence_model(image_np, image_size, task_prompt, text_input=None):
 # Function to plot image with bounding boxes
 def plot_image_with_bboxes(image_np, bboxes, labels=None):
     fig, ax = plt.subplots(1)
-    ax.imshow(image_np)
+    ax.imshow(image_np / 255.0)  # Normalize the image for plotting
     colors = ['red', 'blue', 'green', 'yellow', 'purple', 'cyan']
     for i, bbox in enumerate(bboxes):
         color = colors[i % len(colors)]
@@ -159,4 +159,4 @@ with gr.Blocks(theme='NoCrypt/miku') as demo:
     """
     gr.HTML(footer)
 
-demo.launch()
+demo.launch()