Update app.py

app.py

@@ -46,14 +46,16 @@ 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"))
-        with closing(create_connection(playground)) as conn:
+        with closing(create_connection("wss://ashrafb-arv3s.hf.space/queue/join")) as conn:
             conn.send('{"fn_index":0,"session_hash":""}')
-            conn.send(f'{{"fn_index":0,"data":["{prompt}"],"session_hash":""}}')
+            prompt = 'Summer, lake, rain'
+            conn.send(f'{{"fn_index":0,"data":["{prompt}", 0, "No style"],"session_hash":""}}')
             conn.recv()
             conn.recv()
             conn.recv()
             conn.recv()
             a = conn.recv()
+            print(">> A: ", a)
             photo = json.loads(a)['output']['data'][0]
             photo = photo.replace('data:image/jpeg;base64,', '').replace('data:image/png;base64,', '')
             photo = Image.open(io.BytesIO(base64.decodebytes(bytes(photo, "utf-8"))))