Fix style

huggingface · Nov 1, 2023 · 85c4a6c · 85c4a6c
1 parent 38fd187
commit 85c4a6c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/transformers/models/fuyu/test_pipeline.py b/src/transformers/models/fuyu/test_pipeline.py
@@ -1,12 +1,15 @@
-from transformers import pipeline
 import io
+
 import requests
 from PIL import Image
 
+from transformers import pipeline
+
+
 pipe = pipeline(task="image-to-text", model="adept/fuyu-8b")
 
 url = "https://huggingface.co/datasets/hf-internal-testing/fixtures-captioning/resolve/main/bus.png"
 image = Image.open(io.BytesIO(requests.get(url).content))
 prompt = "Generate a coco-style caption.\n"
 
-results = pipe(image, prompt=prompt)
+results = pipe(image, prompt=prompt)