Update app.py
Browse files
app.py
CHANGED
@@ -30,6 +30,8 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
|
|
30 |
API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/lyrielv16"
|
31 |
if model == 'Animagine XL 2.0':
|
32 |
API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl-2.0"
|
|
|
|
|
33 |
|
34 |
payload = {
|
35 |
"inputs": prompt,
|
@@ -53,7 +55,7 @@ with gr.Blocks(css=css) as dalle:
|
|
53 |
with gr.Row():
|
54 |
with gr.Column(elem_id="prompt-container"):
|
55 |
text_prompt = gr.Textbox(label="Prompt", placeholder="Описание изображения", lines=3, elem_id="prompt-text-input")
|
56 |
-
model = gr.Radio(label="Модель", value="DALL-E 3 XL", choices=["AbsoluteReality v1.8.1", "DALL-E 3 XL", "Playground v2", "Openjourney v4", "Lyriel 1.6", "Animagine XL 2.0"])
|
57 |
|
58 |
|
59 |
with gr.Tab("Расширенные настройки"):
|
|
|
30 |
API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/lyrielv16"
|
31 |
if model == 'Animagine XL 2.0':
|
32 |
API_URL = "https://api-inference.huggingface.co/models/Linaqruf/animagine-xl-2.0"
|
33 |
+
if model == 'Counterfeit v2.5':
|
34 |
+
API_URL = "https://api-inference.huggingface.co/models/gsdf/Counterfeit-V2.5"
|
35 |
|
36 |
payload = {
|
37 |
"inputs": prompt,
|
|
|
55 |
with gr.Row():
|
56 |
with gr.Column(elem_id="prompt-container"):
|
57 |
text_prompt = gr.Textbox(label="Prompt", placeholder="Описание изображения", lines=3, elem_id="prompt-text-input")
|
58 |
+
model = gr.Radio(label="Модель", value="DALL-E 3 XL", choices=["AbsoluteReality v1.8.1", "DALL-E 3 XL", "Playground v2", "Openjourney v4", "Lyriel 1.6", "Animagine XL 2.0", "Counterfeit v2.5"])
|
59 |
|
60 |
|
61 |
with gr.Tab("Расширенные настройки"):
|