Update app.py

app.py

@@ -43,6 +43,19 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     
     url_sd5 = os.getenv("url_sd5")
     url_sd6 = os.getenv("url_sd6")
+
+    if task == "Playground v2":
+        playground = srt(os.getenv("playground"))
+        client = Client(playground)
+        result = client.predict(prompt, "", False, 220, 1024, 1024, 3, True, api_name="/run")
+        return result[0][0]['image']
+
+    if task == "OpenDalle v1.1":
+        opendalle = str(os.getenv("opendalle"))
+        client = Client(opendalle)
+        result = client.predict(prompt, "", "", "", False, False, False, 999, 1024, 1024, 5, 5, 25, 25, False, api_name="/run")
+        return result
+        
     try:
         with closing(create_connection(f"{url_sd3}", timeout=60)) as conn:
             conn.send('{"fn_index":3,"session_hash":""}')
@@ -185,7 +198,8 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             task = gr.Radio(interactive=True, value="Stable Diffusion XL 1.0", show_label=True, label="Модель нейросети:", choices=['Stable Diffusion XL 1.0', 'Crystal Clear XL', 
                                                                                                               'Juggernaut XL', 'DreamShaper XL',
-                                                                                                              'SDXL Niji', 'Cinemax SDXL', 'NightVision XL'])
+                                                                                                              'SDXL Niji', 'Cinemax SDXL', 'NightVision XL',
+                                                                                                              'Playground v2', 'OpenDalle v1.1'])
     with gr.Tab("Расширенные настройки"):
         with gr.Row():
             negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=True, label='Negative Prompt:', lines=3, value="[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry")