Update app.py

app.py

@@ -10,8 +10,8 @@ from langdetect import detect
 
 api_base = os.getenv("API_BASE")
 mmodels = {
-    "DALL-E 3 XL": "openskyml/dalle-3-xl",
-    "Playground 2": "playgroundai/playground-v2-1024px-aesthetic",
+    "1": "openskyml/dalle-3-xl",
+    "2": "playgroundai/playground-v2-1024px-aesthetic",
     "Openjourney 4": "prompthero/openjourney-v4",
     "AbsoluteReality 1.8.1": "digiplay/AbsoluteReality_v1.8.1",
     "Lyriel 1.6": "stablediffusionapi/lyrielv16",
@@ -30,7 +30,7 @@ mmodels = {
     "Redmond SDXL": "artificialguybr/LogoRedmond-LogoLoraForSDXL-V2",
     "Arcane": "nitrosocke/Arcane-Diffusion"
 }
-def query(prompt, is_negative=False, gpt=True, model="DALL-E 3 XL", steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
+def query(prompt, is_negative=False, gpt=True, model="1", steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024):
     if prompt == "" or prompt == None:
         raise gr.Error("🙏 Опиши пожалуйста как должно выглядеть изображение")
         return None
@@ -157,7 +157,8 @@ with gr.Blocks(css=css, theme="YTheme/KidsShool") as dalle:
     with gr.Tab("⚙️ Расширенные настройки"):
         with gr.Row():
             negative_prompt = gr.Textbox(label="🚫 Чего не должно быть на фото?", placeholder="Текст, кошка", value="Искажения, пиксельность, блюр, размыто, плохое качество, текст", lines=3, elem_id="negative-prompt-text-input")
-    
+        with gr.Row():
+            model = gr.Radio(label="Версия нейросети", phoices=["1", "2"], value="1")
         with gr.Row():
             gpt = gr.Checkbox(label="😊 Помогатор (Просить Помогатора улучшить описание?)", value=True)