Update README.md

README.md

@@ -68,10 +68,8 @@ def run_example(task_prompt, image, max_new_tokens=128):
 def plot_bbox(image, data):
    # Create a figure and axes
     fig, ax = plt.subplots()
-
     # Display the image
     ax.imshow(image)
-
     # Plot each bounding box
     for bbox, label in zip(data['bboxes'], data['labels']):
         # Unpack the bounding box coordinates
@@ -82,10 +80,8 @@ def plot_bbox(image, data):
         ax.add_patch(rect)
         # Annotate the label
         plt.text(x1, y1, label, color='white', fontsize=8, bbox=dict(facecolor='red', alpha=0.5))
-
     # Remove the axis ticks and labels
     ax.axis('off')
-
     # Show the plot
     plt.show()
 
@@ -98,7 +94,6 @@ example_id = 5
 image = dataset["train"][example_id]["image"]
 
 parsed_answer = run_example("<OD>", image=image)
-
 plot_bbox(image, parsed_answer["<OD>"])
 ```