imagennnn / app.py
Serg4451D's picture
Update app.py
7cb39a8 verified
import gradio as gr
from g4f.client import Client
# Инициализация клиента
client = Client()
def generate_image(prompt, model):
# Генерация изображения
response = client.images.generate(
model=model,
prompt=prompt
)
# Получение URL сгенерированного изображения
image_url = response.data[0].url
return image_url
# Список доступных моделей
models = ["flux-realism", "sdxl", "any_dark", "dalle-2", "dalle-3"]
# Создание интерфейса Gradio
iface = gr.Interface(
fn=generate_image,
inputs=[
gr.Textbox(label="Введите текстовый запрос"),
gr.Dropdown(choices=models, label="Выберите модель", value="flux")
],
outputs="image",
title="Генератор изображений",
description="Введите текстовый запрос и выберите модель, чтобы сгенерировать изображение."
)
# Запуск интерфейса
iface.launch()