Update app.py
Browse files
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()
|