|
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 = "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 |
|
|
|
|
|
iface = gr.Interface( |
|
fn=generate_image_from_text, |
|
inputs=["text", "slider"], |
|
outputs="image", |
|
title="Text-to-Image Generator", |
|
description="Enter a text prompt to generate an image." |
|
) |
|
|
|
|
|
iface.launch() |
