Update app.py

app.py

@@ -11,12 +11,16 @@ API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
 API_TOKEN = os.getenv("HF_READ_TOKEN") # it is free
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 
-def query(prompt, is_negative=False, steps=1, cfg_scale=6, seed=None):
+def query(prompt, model, is_negative=False, steps=20, cfg_scale=7, seed=None):
     language = detect(prompt)
     
     if language == 'ru':
         prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
         print(prompt)
+    
+    if model == 'DALL-E 3 XL'
+        API_URL = "https://api-inference.huggingface.co/models/openskyml/dalle-3-xl"
+
     payload = {
         "inputs": prompt,
         "is_negative": is_negative,
@@ -39,6 +43,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=1, elem_id="prompt-text-input")
+                model = gr.Radio(label="Модель", value="DALL-E 3 XL", choices=["DALL-E 3 XL"])
                 
 
     with gr.Tab("Расширенные настройки"):
@@ -49,6 +54,6 @@ with gr.Blocks(css=css) as dalle:
     with gr.Row():
         image_output = gr.Image(type="pil", label="Изображение", elem_id="gallery")
         
-    text_button.click(query, inputs=[text_prompt, negative_prompt], outputs=image_output)
+    text_button.click(query, inputs=[text_prompt, model, negative_prompt], outputs=image_output)
 
 dalle.launch(show_api=False)