File size: 1,084 Bytes
69a9435 d306302 0094cc8 d306302 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import gradio as gr
import requests
import numpy as np
from PIL import Image
def enlarge_image(image, scale):
# convert image to bytes
img_byte_arr = image.tobytes()
# make request to DALL-E 2 API
response = requests.post("https://api.dall-e.com/v1/enlarge",
data=img_byte_arr,
params={"size": scale})
# get the image bytes from the response
image_bytes = response.content
# convert bytes to numpy array
img_np = np.array(Image.open(BytesIO(image_bytes)))
return img_np
# create the Gradio interface
input_image = gr.inputs.Image(label="Input Image")
scale = gr.inputs.Number(label="Scale", default=2)
output_image = gr.outputs.Image(label="Enlarged Image")
title = "DALL-E 2 Image Enlarger"
description = "Enlarges an image using DALL-E 2. Enter a scale factor and upload an image."
examples = [["examples/butterfly.jpg"]]
gr.Interface(fn=enlarge_image, inputs=[input_image, scale], outputs=output_image,
title=title, description=description, examples=examples).launch()
|