Rooni commited on
Commit
f158749
1 Parent(s): 082714d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -8,6 +8,8 @@ url = "https://stablediffusionapi.com/api/v4/dreambooth"
8
  # API ключ
9
  api_key = os.getenv("KEY")
10
 
 
 
11
 
12
  # Функция для отправки запроса
13
  def render(model, prompt, negative_prompt, width, height, guidance, seed, upscale):
@@ -43,7 +45,7 @@ def render(model, prompt, negative_prompt, width, height, guidance, seed, upscal
43
  gr.Interface(
44
  render,
45
  inputs=[
46
- gr.Dropdown(["realistic-vision-51"], label="Модель", value="realistic-vision-51"),
47
  gr.Textbox(label="Описание изображения:", placeholder="Введите описание изображения", lines=3),
48
  gr.Textbox(label="Negative Prompt:", placeholder="Введите Negative Prompt", value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry"),
49
  gr.Slider(show_label=True, minimum=256, maximum=2048, label="Ширина:", value="512", step=1),
 
8
  # API ключ
9
  api_key = os.getenv("KEY")
10
 
11
+ model_list = ["realistic-vision-51", "anything-v5"]
12
+
13
 
14
  # Функция для отправки запроса
15
  def render(model, prompt, negative_prompt, width, height, guidance, seed, upscale):
 
45
  gr.Interface(
46
  render,
47
  inputs=[
48
+ gr.Dropdown(model_list, label="Модель", value="realistic-vision-51"),
49
  gr.Textbox(label="Описание изображения:", placeholder="Введите описание изображения", lines=3),
50
  gr.Textbox(label="Negative Prompt:", placeholder="Введите Negative Prompt", value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry"),
51
  gr.Slider(show_label=True, minimum=256, maximum=2048, label="Ширина:", value="512", step=1),