feat: Add check_requirements for "transformers" in HuggingFaceVideoClassifier

action-recog
fcakyon 7 months ago
parent 9b8ddb7021
commit 5b23686d84
  1. 3
      ultralytics/solutions/action_recognition.py

@ -7,7 +7,7 @@ import cv2
import numpy as np
import torch
from ultralytics.utils.checks import check_imshow
from ultralytics.utils.checks import check_imshow, check_requirements
from ultralytics.utils.plotting import Annotator
from ultralytics.utils.torch_utils import select_device
@ -358,6 +358,7 @@ class HuggingFaceVideoClassifier:
device (str or torch.device, optional): The device to run the model on. Defaults to "".
fp16 (bool, optional): Whether to use FP16 for inference. Defaults to False.
"""
check_requirements("transformers")
from transformers import AutoModel, AutoProcessor
self.fp16 = fp16

Loading…
Cancel
Save