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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -24,12 +24,8 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
24
  API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
25
  if model == 'Openjourney v4':
26
  API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney-v4"
27
- if model == 'Anything V5':
28
- API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/anything-v5"
29
  if model == 'AbsoluteReality v1.8.1':
30
  API_URL = "https://api-inference.huggingface.co/models/digiplay/AbsoluteReality_v1.8.1"
31
- if model == 'Dreamshaper v8':
32
- API_URL = "https://api-inference.huggingface.co/models/Lykon/dreamshaper-8"
33
  if model == 'Lyriel 1.6':
34
  API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/lyrielv16"
35
 
@@ -55,7 +51,7 @@ with gr.Blocks(css=css) as dalle:
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", "Anything V5", "Dreamshaper v8", "Lyriel 1.6"])
59
 
60
 
61
  with gr.Tab("Расширенные настройки"):
 
24
  API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
25
  if model == 'Openjourney v4':
26
  API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney-v4"
 
 
27
  if model == 'AbsoluteReality v1.8.1':
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
 
 
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("Расширенные настройки"):