prepush

Author: Bryannah Hernandez

src/sagemaker/app.py

@@ -35,7 +35,7 @@ async def generate_text(prompt: Request):
     return generated_text[0]["generated_text"]
 
 
-generator = pipeline('text-generation', model='gpt2')
+generator = pipeline("text-generation", model="gpt2")
 
 
 @app.post("/post")

src/sagemaker/pipeline.py

@@ -45,6 +45,7 @@
 
 logger = logging.getLogger(__name__)
 
+
 class PipelineModel(object):
     """A pipeline of SageMaker `Model` instances.