still fixing json issue

damilojohn · Aug 15, 2023 · afbb558 · afbb558
1 parent 6de04d8
commit afbb558
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lambda_handler.py b/lambda_handler.py
@@ -32,7 +32,7 @@ def GetEmbeddingsFromS3():
 def handler(event, _context):
     """main model prediction api"""
     data = json.loads(event)
-    prompt = data.get('prompt')
+    prompt = data["prompt"]
     if event is None:
         return {'statusCode': 400, 'message': 'no input prompt was provided'}
     logger.info('downloading embeddings from s3....')