flux-1.1-pro / app.py
Serg4451D's picture
Update app.py
dcd4b12 verified
import gradio as gr
from g4f.client import Client
def generate_image(prompt):
try:
client = Client()
response = client.images.generate(
model="flux-realism",
prompt=prompt,
# Можете добавить дополнительные параметры
)
# Получаем URL изображения
image_url = response.data[0].url
return image_url
except Exception as e:
return f"Ошибка при генерации изображения: {str(e)}"
# Создаем интерфейс Gradio
def create_image_generator():
with gr.Blocks() as demo:
gr.Markdown("# Генератор изображений с помощью AI")
with gr.Row():
# Текстовое поле для ввода промпта
prompt_input = gr.Textbox(label="Опишите изображение")
# Кнопка генерации
generate_btn = gr.Button("Сгенерировать изображение")
# Область для отображения сгенерированного изображения
output_image = gr.Image(label="Сгенерированное изображение")
# Привязываем функцию генерации к кнопке
generate_btn.click(
fn=generate_image,
inputs=prompt_input,
outputs=output_image
)
return demo
# Запуск приложения
if __name__ == "__main__":
demo = create_image_generator()
demo.launch(
share=True, # Создает публичную ссылку
debug=True # Включает режим отладки
)