Update README.md

README.md

@@ -29,7 +29,6 @@ It achieves the following results on the evaluation set:
     processor = AutoProcessor.from_pretrained("Musa07/Florence-2-large-FormClassification-ft", trust_remote_code=True)
   
     def run_example(task_prompt, image, max_new_tokens=128):
-  
       prompt = task_prompt
       inputs = processor(text=prompt, images=image, return_tensors="pt")
       generated_ids = model.generate(
@@ -47,9 +46,8 @@ It achieves the following results on the evaluation set:
           image_size=(image.width, image.height)
       )
       return parsed_answer
-  
-  def plot_bbox(image, data):
-  
+      
+    def plot_bbox(image, data):
       fig, ax = plt.subplots()
   
       # Display the image
@@ -72,10 +70,10 @@ It achieves the following results on the evaluation set:
       # Show the plot
       plt.show()
       
-  image = Image.open('1.jpeg')
-  parsed_answer = run_example("<OD>", image=image)
-  print(parsed_answer)
-  plot_bbox(image, parsed_answer["<OD>"])
+    image = Image.open('1.jpeg')
+    parsed_answer = run_example("<OD>", image=image)
+    print(parsed_answer)
+    plot_bbox(image, parsed_answer["<OD>"])