app.py

import streamlit as st
import os
import time
import json
import requests
from PIL import Image
from io import BytesIO
import base64

def dalle(image, model):
    start = time.time()
    im = Image.open(BytesIO(base64.b64decode(image)))
    im = im.resize((224, 224))
    img_array = np.asarray(im)
    img_array = np.expand_dims(img_array, axis=0)
    response = model.predict(img_array)
    elapsed_time = time.time() - start
    result = {'time':elapsed_time, 'prediction':response[0].tolist()}
    return json.dumps(result)