app.py

import gradio as gr
import requests
import os

# Замените "YOUR_MODEL_ENDPOINT_HERE" и "YOUR_BEARER_TOKEN_HERE" на фактический адрес вашей модели Hugging Face и ваш Bearer токен
MODEL_ENDPOINT = os.getenv("BASE_URL")
BEARER_TOKEN = os.getenv("API_KEY")

def generate_image(prompt, negative_prompt, sampling_steps, cfg_scale, seed):
    # Создаем заголовок авторизации Bearer
    headers = {"Authorization": f"Bearer {BEARER_TOKEN}"}

    # Отправляем запрос к API модели Hugging Face с использованием введенных параметров и заголовка авторизации
    response = requests.post(
        MODEL_ENDPOINT,
        json={
            "prompt": prompt,
            "negative_prompt": negative_prompt,
            "sampling_steps": sampling_steps,
            "cfg_scale": cfg_scale,
            "seed": seed
        },
        headers=headers
    )

    # Получаем сгенерированное изображение из ответа API
    generated_image = response.content

    return generated_image

iface = gr.Interface(
    fn=generate_image,
    inputs=[
        gr.Textbox("text", label="Prompt"),
        gr.Textbox("text", label="Negative Prompt"),
        gr.Slider(minimum=1, maximum=30, label="Sampling Steps"),
        gr.Slider(minimum=0.1, maximum=2.0, label="CFG Scale"),
        gr.Textbox("number", label="Seed (default -1)")
    ],
    outputs=gr.Image(label="Generated Image"),
    live=True,
    title="Hugging Face Image Generator",
    theme="compact",
    layout="vertical"
)

iface.launch()