25 lines
738 B
Python
25 lines
738 B
Python
import json
|
|
import base64
|
|
# from PIL import Image
|
|
import io
|
|
from model_handler import ModelHandler
|
|
|
|
def init_context(context):
|
|
context.logger.info("Init context... 0%")
|
|
|
|
context.logger.info("Initializing EMDetection model...")
|
|
context.user_data.model_handler = ModelHandler()
|
|
|
|
context.logger.info("Init context...100%")
|
|
|
|
def handler(context, event):
|
|
context.logger.info("Run EMDetection model")
|
|
data = event.body
|
|
image_data = base64.b64decode(data["image"])
|
|
threshold = float(data.get("threshold", 0.5))
|
|
|
|
results = context.user_data.model_handler.infer(image_data, threshold)
|
|
|
|
return context.Response(body=json.dumps(results), headers={},
|
|
content_type='application/json', status_code=200)
|