Rooni commited on
Commit
db91780
1 Parent(s): 786fb4f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -28,6 +28,8 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
28
  API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
29
  if model == 'Lyriel 1.6':
30
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/lyrielv16"
 
 
31
 
32
  payload = {
33
  "inputs": prompt,
@@ -51,7 +53,7 @@ with gr.Blocks(css=css) as dalle:
51
  with gr.Row():
52
  with gr.Column(elem_id="prompt-container"):
53
  text_prompt = gr.Textbox(label="Prompt", placeholder="Описание изображения", lines=3, elem_id="prompt-text-input")
54
- 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"])
55
 
56
 
57
  with gr.Tab("Расширенные настройки"):
 
28
  API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
29
  if model == 'Lyriel 1.6':
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
  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("Расширенные настройки"):