Rooni commited on
Commit
260ef11
1 Parent(s): db91780

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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("Расширенные настройки"):