Rooni commited on
Commit
b6b49e4
·
1 Parent(s): e54d489

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -7
app.py CHANGED
@@ -71,13 +71,17 @@ app = gr.Interface(
71
  generate_image,
72
  title="Gradio Image Generator",
73
  description="Генератор изображений с помощью ИИ",
74
- inputs={
75
- "prompt": gr.Input(type="text", label="Текстовый запрос"),
76
- "negative_prompt": gr.Input(type="text", label="Отрицательный текстовый запрос"),
77
- "model": gr.Select(models, label="Модель"),
78
- },
79
- outputs={"image": gr.Output(type="image", label="Сгенерированное изображение")},
80
- buttons={"Улучшить качество": improve_image},
 
 
 
 
81
  )
82
 
83
  if __name__ == "__main__":
 
71
  generate_image,
72
  title="Gradio Image Generator",
73
  description="Генератор изображений с помощью ИИ",
74
+ inputs=[
75
+ gr.inputs.Textbox(label="Текстовый запрос"),
76
+ gr.inputs.Textbox(label="Отрицательный текстовый запрос"),
77
+ gr.inputs.Dropdown(choices=models, label="Модель"),
78
+ ],
79
+ outputs=[
80
+ gr.outputs.Image(label="Сгенерированное изображение")
81
+ ],
82
+ buttons=[
83
+ gr.buttons.Button("Улучшить качество", fn=improve_image)
84
+ ]
85
  )
86
 
87
  if __name__ == "__main__":