Update app.py

app.py

@@ -92,12 +92,13 @@ def next_image():
             print(prompt)
 
             im_emb_st = str(im_emb[0].cpu().detach().tolist())[1:-1]
-            output = StringIO()
-            output.write(im_emb_st)
+            
+            #output = StringIO()
+            #output.write(im_emb_st)
 
             output = replicate.run(
             "rynmurdock/zahir:210b8cdb9aa8ac97325fd534db13ad1a2b3b1226ee2109a20952a810b94216b6",
-            input={"prompt": prompt, 'im_emb': output}
+            input={"prompt": prompt, 'im_emb': im_emb_st}
             )
             response = requests.get(output['file1'])
             image = Image.open(BytesIO(response.content))
@@ -140,8 +141,6 @@ def choose(choice):
 
 css = "div#output-image {height: 768px !important; width: 768px !important; margin:auto;}"
 with gr.Blocks(css=css) as demo:
-    with gr.Row():
-        html = gr.HTML('''<div style='text-align:center; font-size:32'>You will callibrate for several prompts and then roam.</ div>''')
     with gr.Row(elem_id='output-image'):
         img = gr.Image(interactive=False, elem_id='output-image',)
     with gr.Row(equal_height=True):
@@ -168,5 +167,7 @@ with gr.Blocks(css=css) as demo:
         b4.click(start,
                  [b4],
                  [b1, b2, b3, b4, img,])
+    with gr.Row():
+        html = gr.HTML('''<div style='text-align:center; font-size:32'>You will callibrate for several prompts and then roam.</ div>''')
 
 demo.launch()  # Share your demo with just 1 extra parameter 🚀