import transformers

Author: Bryannah Hernandez

requirements/extras/huggingface_requirements.txt

@@ -3,4 +3,3 @@ sagemaker_schema_inference_artifacts>=0.0.5
 uvicorn>=0.30.1
 fastapi>=0.111.0
 nest-asyncio
-transformers

requirements/extras/test_requirements.txt

@@ -43,4 +43,3 @@ huggingface_hub>=0.23.4
 uvicorn>=0.30.1
 fastapi>=0.111.0
 nest-asyncio
-transformers

src/sagemaker/serve/app.py

@@ -5,6 +5,7 @@
 import logging
 import importlib.util
 import uvicorn
+import transformers
 
 from transformers import pipeline
 from fastapi import FastAPI, Request