Spaces:
Sleeping
Sleeping
Clement Vachet
commited on
Commit
·
bc42f94
1
Parent(s):
a1a1640
Comment out query string parameters
Browse files- lambda_function.py +17 -9
lambda_function.py
CHANGED
|
@@ -38,22 +38,30 @@ def detection_pipeline(model_type, image_bytes):
|
|
| 38 |
|
| 39 |
|
| 40 |
def lambda_handler(event, context):
|
|
|
|
| 41 |
try:
|
| 42 |
# Get the model name from the query string parameters
|
| 43 |
# Condition for local testing
|
| 44 |
-
is_querystringparam = event.get('queryStringParameters')
|
| 45 |
-
if is_querystringparam is not None:
|
| 46 |
-
|
| 47 |
-
else:
|
| 48 |
-
|
| 49 |
-
model_type = get_model_type(model_query)
|
| 50 |
-
logger.info(f"Model query: {model_query}")
|
| 51 |
-
logger.info(f"Model type: {model_type}")
|
|
|
|
|
|
|
| 52 |
|
| 53 |
# Decode the base64-encoded image data from the event
|
| 54 |
-
image_data =
|
|
|
|
|
|
|
|
|
|
|
|
|
| 55 |
result_dict = detection_pipeline(model_type, image_data)
|
| 56 |
logger.info(f"API Results: {result_dict}")
|
|
|
|
| 57 |
return {
|
| 58 |
'statusCode': 200,
|
| 59 |
'headers': {
|
|
|
|
| 38 |
|
| 39 |
|
| 40 |
def lambda_handler(event, context):
|
| 41 |
+
logger.info(f"API event: {event}")
|
| 42 |
try:
|
| 43 |
# Get the model name from the query string parameters
|
| 44 |
# Condition for local testing
|
| 45 |
+
# is_querystringparam = event.get('queryStringParameters')
|
| 46 |
+
# if is_querystringparam is not None:
|
| 47 |
+
# model_query = event['queryStringParameters'].get('model', '').lower()
|
| 48 |
+
# else:
|
| 49 |
+
# model_query = ""
|
| 50 |
+
# model_type = get_model_type(model_query)
|
| 51 |
+
# logger.info(f"Model query: {model_query}")
|
| 52 |
+
# logger.info(f"Model type: {model_type}")
|
| 53 |
+
|
| 54 |
+
model_type = get_model_type("")
|
| 55 |
|
| 56 |
# Decode the base64-encoded image data from the event
|
| 57 |
+
image_data = event['body']
|
| 58 |
+
if event['isBase64Encoded']:
|
| 59 |
+
image_data = base64.b64decode(image_data)
|
| 60 |
+
|
| 61 |
+
# Run detection pipeline
|
| 62 |
result_dict = detection_pipeline(model_type, image_data)
|
| 63 |
logger.info(f"API Results: {result_dict}")
|
| 64 |
+
|
| 65 |
return {
|
| 66 |
'statusCode': 200,
|
| 67 |
'headers': {
|