apapiu/transformer_diffusion_gpu_api

1import gradio as gr2import requests3from PIL import Image4from io import BytesIO5import os6 7runpod_id = os.environ['RUNPOD_ID']8token_id = os.environ['AUTH_TOKEN']9 10#url = f'https://{runpod_id}-8000.proxy.runpod.net/generate-image/'11 12url = "http://my-tld-app-container-dns.eastus.azurecontainer.io/generate-image/"13 14 15def generate_image_from_text(prompt, class_guidance):16    headers = {17        'Authorization': f'Bearer {token_id}'18    }19    20    data = {21        "prompt": prompt,22        "class_guidance": class_guidance,23        "seed": 11,24        "num_imgs": 1,25        "img_size": 3226    }27    28    response = requests.post(url, json=data, headers=headers)29    30    if response.status_code == 200:31        image = Image.open(BytesIO(response.content))32    else:33        print("Failed to fetch image:", response.status_code, response.text)34 35    return image36 37# Define the Gradio interface38iface = gr.Interface(39    fn=generate_image_from_text,  # The function to generate the image40    inputs=["text", "slider"],41    outputs="image",42    title="Text-to-Image Generator",43    description="Enter a text prompt to generate an image."44)45 46# Launch the app47iface.launch()