Vertex AI Prediction example

app.py

@app.route('/predict',methods=['GET','POST'])
def predict():
  # Decode image
  request_json = request.get_json(silent=True, force=True)
  data = request_json['instances']
  decoded = base64.urlsafe_b64decode(data[0]["b64"])
  img_in_bytes = io.BytesIO(decoded)
  img = Image.open(img_in_bytes).convert('RGB')
  
  # Generate Anime image
  img_anime = face2paint(model=model2, img=img)

  # Encode image for response
  img_bytes = io.BytesIO()
  img_anime.save(img_bytes, format='JPEG')
  enc_out = base64.urlsafe_b64encode(img_bytes.getvalue())
  return json.dumps({"predictions": [{"b64": enc_out}]})