Update app.py

app.py

@@ -2,20 +2,20 @@ import gradio as gr
 import requests
 import random
 
-def generate_image(prompt, model, width, height, seed):
+def generate_image(prompt, width, height, seed):
     if seed == -1:
         seed = random.randint(0, 999999)
     url = f"https://image.pollinations.ai/prompt/{prompt}?width={width}&height={height}&seed={seed}&nologo=true&nofeed=true"
-    if model:
-        url += f"&model={model}"
     response = requests.get(url)
-    return response.content
+    if response.status_code == 200:
+        return response.content
+    else:
+        return None
 
-# Основные настройки
+# Создаем интерфейс Gradio
 with gr.Blocks() as demo:
     with gr.Tab("Основные настройки"):
         prompt_input = gr.Textbox(label="Промпт", placeholder="Введите промпт на английском языке")
-        model_dropdown = gr.Dropdown(label="Выбор модели", choices=["Flux", "Flux-Pro", "Flux-Realism", "Private", "Flux-Anime", "Flux-3D", "Flux-CablyAl", "Turbo"], value="Flux")
     
     with gr.Tab("Дополнительные настройки"):
         width_slider = gr.Slider(label="Ширина", minimum=1, maximum=4000, value=1024)
@@ -23,8 +23,8 @@ with gr.Blocks() as demo:
         seed_slider = gr.Slider(label="Сид", minimum=-1, maximum=999999, value=-1, step=1)
     
     create_button = gr.Button("Создать")
-    output_image = gr.Image(label="Изображение")
+    output_image = gr.Image(label="Изображение", type="pil")  # Указываем тип вывода как PIL изображение
 
-    create_button.click(fn=generate_image, inputs=[prompt_input, model_dropdown, width_slider, height_slider, seed_slider], outputs=output_image)
+    create_button.click(fn=generate_image, inputs=[prompt_input, width_slider, height_slider, seed_slider], outputs=output_image)
 
-demo.launch()
+demo.launch()