fastapi

Author: Bryannah Hernandez

src/sagemaker/serve/app.py

@@ -3,11 +3,16 @@
 from __future__ import absolute_import
 
 import logging
-from fastapi import FastAPI, Request
 
 
 logger = logging.getLogger(__name__)
 
+try:
+    from fastapi import FastAPI, Request
+
+except ImportError:
+    logger.error("To enable in_process mode for Transformers install fastapi from HuggingFace hub")
+
 app = FastAPI(
     title="Transformers In Process Server",
     version="1.0",