File size: 1,171 Bytes
78c2594 eeba8b2 78c2594 da56136 7013fe5 da56136 e8e6426 e7e8042 6ba245d da56136 28a9b53 da56136 7013fe5 da56136 7641117 da56136 78c2594 eeba8b2 |
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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
import gradio as gr
import requests
from PIL import Image
from io import BytesIO
import os
runpod_id = os.environ['RUNPOD_ID']
token_id = os.environ['AUTH_TOKEN']
#url = f'https://{runpod_id}-8000.proxy.runpod.net/generate-image/'
url = "http://my-tld-app-container-dns.eastus.azurecontainer.io/generate-image/"
def generate_image_from_text(prompt, class_guidance):
headers = {
'Authorization': f'Bearer {token_id}'
}
data = {
"prompt": prompt,
"class_guidance": class_guidance,
"seed": 11,
"num_imgs": 1,
"img_size": 32
}
response = requests.post(url, json=data, headers=headers)
if response.status_code == 200:
image = Image.open(BytesIO(response.content))
else:
print("Failed to fetch image:", response.status_code, response.text)
return image
# Define the Gradio interface
iface = gr.Interface(
fn=generate_image_from_text, # The function to generate the image
inputs=["text", "slider"],
outputs="image",
title="Text-to-Image Generator",
description="Enter a text prompt to generate an image."
)
# Launch the app
iface.launch() |