Skip to content

Instantly share code, notes, and snippets.

@francoismarceau29

francoismarceau29/flask_template_aws.py

Last active July 18, 2017 23:48
Show Gist options
  • Save francoismarceau29/a9e0164f8414a994aeaa1b2a77efabc1 to your computer and use it in GitHub Desktop.
Save francoismarceau29/a9e0164f8414a994aeaa1b2a77efabc1 to your computer and use it in GitHub Desktop.
Download ZIP
Basic Flask template for AWS Lambda
Raw
flask_template_aws.py
from sklearn.externals import joblib
from boto.s3.key import Key
from boto.s3.connection import S3Connection
from flask import Flask
from flask import request
from flask import json
BUCKET_NAME = 'your-s3-bucket-name'
MODEL_FILE_NAME = 'your-model-name.pkl'
MODEL_LOCAL_PATH = '/tmp/' + MODEL_FILE_NAME
app = Flask(__name__)
@app.route('/', methods=['POST'])
def index():
payload = json.loads(request.get_data().decode('utf-8'))
prediction = predict(payload['payload'])
data = {}
data['data'] = prediction[-1]
return json.dumps(data)
def load_model():
print 'Loading model from S3'
def predict(data):
print 'Making predictions'
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment