Update app.py

app.py

@@ -22,8 +22,16 @@ def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
         API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
     if model == 'Playground v2':
         API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2-1024px-aesthetic"
-    if model == 'Openjourney':
-        API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney" 
+    if model == 'Openjourney v4':
+        API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney-v4"
+    if model == 'Anything V5':
+        API_URL = "https://api-inference.huggingface.co/models/hogiahien/anything-v5-edited"
+    if model == 'AbsoluteReality v1.8.1':
+        API_URL = "Dreamshaper v8"
+    if model == 'https://api-inference.huggingface.co/models/Lykon/dreamshaper-8':
+        API_URL = "link"
+    if model == 'Lyriel 1.6':
+        API_URL = "https://api-inference.huggingface.co/models/stablediffusionapi/lyrielv16"
 
     payload = {
         "inputs": prompt,
@@ -47,7 +55,7 @@ with gr.Blocks(css=css) as dalle:
         with gr.Row():
             with gr.Column(elem_id="prompt-container"):
                 text_prompt = gr.Textbox(label="Prompt", placeholder="Описание изображения", lines=3, elem_id="prompt-text-input")
-                model = gr.Radio(label="Модель", value="DALL-E 3 XL", choices=["DALL-E 3 XL", "Playground v2", "Openjourney"])
+                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"])
                 
 
     with gr.Tab("Расширенные настройки"):