Update app.py

app.py

@@ -55,8 +55,11 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             conn.recv()
             a = conn.recv()
             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"))))
             return photo
             
+            
     if task == "Artigen v3":
         artigen = str(os.getenv("artigen"))
         with closing(create_connection(artigen)) as conn:
@@ -68,6 +71,8 @@ def flip_text(prompt, negative_prompt, task, steps, sampler, cfg_scale, seed):
             conn.recv()
             a = conn.recv()
             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"))))
             return photo
         
     try: