Fix: linter format

geminixiang · Sep 3, 2023 · 4cc1eff · 4cc1eff
1 parent e360d31
commit 4cc1eff
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/huggingface/__init__.py b/huggingface/__init__.py
@@ -1,2 +0,0 @@
-from .blip_image_captioning_large import blip_image_captioning_large
-from .yolos_tiny import yolos_tiny

diff --git a/huggingface/blip_image_captioning_large.py b/huggingface/blip_image_captioning_large.py
@@ -1,5 +1,4 @@
 import requests
-import torch
 from loguru import logger
 from PIL import Image
 from transformers import BlipForConditionalGeneration, BlipProcessor

diff --git a/huggingface/yolos_tiny.py b/huggingface/yolos_tiny.py
@@ -19,14 +19,19 @@ def yolos_tiny(url: str) -> ObjectDetectionResult:
     Returns:
         ObjectDetectionResult: object detection result
     """
-    model = YolosForObjectDetection.from_pretrained("hustvl/yolos-tiny").to(settings.compute_mode)
+    model = YolosForObjectDetection.from_pretrained(
+        "hustvl/yolos-tiny"
+    ).to(settings.compute_mode)
     image_processor = YolosImageProcessor.from_pretrained(
         "hustvl/yolos-tiny", torch_dtype=torch.float16
     )
 
     image = Image.open(requests.get(url, stream=True).raw)
     logger.info(f"Downloaded image from {url}")
-    inputs = image_processor(images=image, return_tensors="pt").to(settings.compute_mode)  # type: ignore
+    inputs = image_processor(
+        images=image,
+        return_tensors="pt"
+    ).to(settings.compute_mode)  # type: ignore
     outputs = model(**inputs)  # type: ignore
 
     target_sizes = torch.tensor([image.size[::-1]])
Original file line number	Diff line number	Diff line change
		@@ -1,2 +0,0 @@
		from .blip_image_captioning_large import blip_image_captioning_large
		from .yolos_tiny import yolos_tiny