| 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() |
