changes

Author: Bryannah Hernandez

requirements/extras/huggingface_requirements.txt

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

src/sagemaker/pipeline.py

@@ -15,7 +15,6 @@
 
 from typing import Optional, Dict, List, Union
 
-import logging
 import sagemaker
 from sagemaker import ModelMetrics, Model
 from sagemaker.config import (
@@ -43,7 +42,6 @@
 from sagemaker.workflow.pipeline_context import runnable_by_pipeline
 from sagemaker.utils import instance_supports_kms
 
-logger = logging.getLogger(__name__)
 
 
 class PipelineModel(object):
@@ -300,7 +298,6 @@ def deploy(
             )
 
         if self.predictor_cls:
-            logger.info("DO I come here")
             predictor = self.predictor_cls(self.endpoint_name, self.sagemaker_session)
             if serializer:
                 predictor.serializer = serializer

src/sagemaker/serve/app.py

@@ -3,10 +3,8 @@
 from __future__ import absolute_import
 
 import logging
-import uvicorn
-import pipeline
-import FastAPI
-import Request
+from transformers import pipeline
+from fastapi import FastAPI, Request
 
 
 logger = logging.getLogger(__name__)