Update app.py

app.py

@@ -46,29 +46,29 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
     hf_token = os.getenv("hf_token")
     if task == "Playground v2":
         playground = str(os.getenv("playground"))
-        client = Client(playground, hf_token=hf_token)
-        result = client.predict(prompt, "", False, 220, 1024, 1024, 3, True, api_name="/run")
-        return result[0][0]['image']
-        
-    if task == "OpenDalle v1.1":
-        headers = {
-            'authority': 'mrfakename-opendallev1-1-gpu-demo.hf.space',
-            'accept': 'text/event-stream',
-            'accept-language': 'ru,en;q=0.9,la;q=0.8,ja;q=0.7',
-            'cache-control': 'no-cache',
-            'referer': 'https://mrfakename-opendallev1-1-gpu-demo.hf.space/?__theme=light',
-            'sec-ch-ua': '"Not_A Brand";v="8", "Chromium";v="120", "YaBrowser";v="24.1", "Yowser";v="2.5"',
-            'sec-ch-ua-mobile': '?0',
-            'sec-ch-ua-platform': '"Windows"',
-            'sec-fetch-dest': 'empty',
-            'sec-fetch-mode': 'cors',
-            'sec-fetch-site': 'same-origin',
-            'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 YaBrowser/24.1.0.0 Safari/537.36',
-        }
-        opendalle = str(os.getenv("opendalle"))
-        client = Client(opendalle, headers=headers)
-        result = client.predict(prompt, "", "", "", False, False, False, 999, 1024, 1024, 5, 5, 25, 25, False, api_name="/run")
-        return result
+        with closing(create_connection(playground)) as conn:
+            conn.send('{"fn_index":0,"session_hash":""}')
+            conn.send(f'{{"fn_index":0,"data":["{prompt}"],"session_hash":""}}')
+            conn.recv()
+            conn.recv()
+            conn.recv()
+            conn.recv()
+            a = conn.recv()
+            photo = json.loads(a)['output']['data'][0]
+            return photo
+            
+    if task == "Artigen v3":
+        artigen = str(os.getenv("artigen"))
+        with closing(create_connection(artigen)) as conn:
+            conn.send('{"fn_index":0,"session_hash":""}')
+            conn.send(f'{{"fn_index":0,"data":["{prompt}", 0, "No style"],"session_hash":""}}')
+            conn.recv()
+            conn.recv()
+            conn.recv()
+            conn.recv()
+            a = conn.recv()
+            photo = json.loads(a)['output']['data'][0]
+            return photo
         
     try:
         with closing(create_connection(f"{url_sd3}", timeout=60)) as conn:
@@ -216,7 +216,7 @@ with gr.Blocks(css=css) as demo:
             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',
-                                                                                                              'Playground v2', 'OpenDalle v1.1'])
+                                                                                                              'Playground v2', 'Artigen v3'])
     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")
@@ -252,5 +252,5 @@ with gr.Blocks(css=css) as demo:
         image_i2i = gr.Image(show_label=True, label='Увеличенное изображение:')
         img2img_b.click(mirror, inputs=[image_output, scale_by, method, gfpgan, codeformer], outputs=image_i2i)
     
-demo.queue(concurrency_count=12)
+demo.queue(concurrency_count=24)
 demo.launch()