Rooni commited on
Commit
1f2d149
·
1 Parent(s): 2679716

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -11
app.py CHANGED
@@ -7,7 +7,7 @@ BASE_URL = os.getenv("BASE_URL")
7
  API_KEY = os.getenv("API_KEY")
8
 
9
  # Функция для отправки запроса к модели
10
- def generate_image(prompt, neg_prompt, sampling_steps, cfg_scale, seed):
11
  # Подготовка данных для запроса
12
  data = {
13
  "prompt": prompt,
@@ -36,6 +36,9 @@ iface = gr.Interface(
36
  inputs=[
37
  gr.Textbox("text", label="Prompt"),
38
  gr.Textbox("text", label="Negative Prompt"),
 
 
 
39
  ],
40
  outputs=gr.Image(),
41
  live=True,
@@ -43,15 +46,5 @@ iface = gr.Interface(
43
  theme="compact"
44
  )
45
 
46
- # Добавление второй вкладки
47
- iface.add_tab(
48
- "Advanced Settings",
49
- [
50
- gr.Slider(minimum=1, maximum=30, default=1, label="Sampling Steps"),
51
- gr.Slider(minimum=1, maximum=20, default=1, label="CFG Scale"),
52
- gr.Number(default=-1, label="Seed")
53
- ]
54
- )
55
-
56
  # Запуск веб-приложения
57
  iface.launch()
 
7
  API_KEY = os.getenv("API_KEY")
8
 
9
  # Функция для отправки запроса к модели
10
+ def generate_image(prompt, neg_prompt, sampling_steps=1, cfg_scale=1, seed=-1):
11
  # Подготовка данных для запроса
12
  data = {
13
  "prompt": prompt,
 
36
  inputs=[
37
  gr.Textbox("text", label="Prompt"),
38
  gr.Textbox("text", label="Negative Prompt"),
39
+ gr.Slider(minimum=1, maximum=30, default=1, label="Sampling Steps"),
40
+ gr.Slider(minimum=1, maximum=20, default=1, label="CFG Scale"),
41
+ gr.Number(default=-1, label="Seed")
42
  ],
43
  outputs=gr.Image(),
44
  live=True,
 
46
  theme="compact"
47
  )
48
 
 
 
 
 
 
 
 
 
 
 
49
  # Запуск веб-приложения
50
  iface.launch()