Spaces:
Sleeping
Sleeping
import io | |
import gradio as gr | |
import requests | |
import base64 | |
from PIL import Image | |
# define the function that will be called when the user inputs text | |
def get_images(text): | |
headers = {'Content-Type': 'application/json'} | |
params = { | |
'return-images': 'true', | |
'number-results': '4', | |
} | |
response = requests.post( | |
"https://wjdr33c1id.execute-api.eu-west-1.amazonaws.com/dev/prediction", | |
params=params, | |
headers=headers, | |
json={"data": text} | |
) | |
# get the list of image data from the response | |
image_data = response.json()["image"] | |
# decode the base64-encoded image data and convert it to PIL images | |
images = [ | |
Image.open(io.BytesIO(base64.b64decode(data))) for data in image_data | |
] | |
# first comes on the top | |
images.reverse() | |
# return the list of images | |
return images | |
# create the gradio app, passing the function as the input and output | |
app = gr.Interface(get_images, | |
gr.components.Textbox(label="Description"), | |
# gr.components.Image(label="Images", type="pil") | |
gr.Gallery(label="Images") | |
) | |
# start the app | |
app.launch() |