Update app.py

app.py

@@ -46,10 +46,13 @@ def infer(text,title):
      Please make sure not to include text in the image."""
     image_url_openai, revised_prompt = generate_image_openai(prompt)
     style_image_url = style_transfer(image_url_openai, f'./Style.png',revised_prompt)
+
+    response_dalle = requests.get(image_url_openai)
+    dalle_img = Image.open(BytesIO(response.content))
     response = requests.get(style_image_url)
     img = Image.open(BytesIO(response.content))
 
-    return img
+    return dalle_img, img
     
     
     
@@ -62,12 +65,12 @@ examples = [
     # "A delicious ceviche cheesecake slice",
 ]
 
-css="""
-#col-container {
-    margin: 0 auto;
-    max-width: 520px;
-}
-"""
+# css="""
+# #col-container {
+#     margin: 0 auto;
+#     max-width: 520px;
+# }
+# """
 
 
 
@@ -84,19 +87,20 @@ with gr.Blocks(css=css) as demo:
                 label="Text",
                 show_label=False,
                 max_lines=1,
-                placeholder="Enter your prompt",
+                placeholder="Enter the `text`",
                 container=False,
             )
             title = gr.Text(
                 label="Title",
                 show_label=False,
                 max_lines=1,
-                placeholder="Enter your prompt",
+                placeholder="Enter the `title`",
                 container=False,
             )
             
             run_button = gr.Button("Run", scale=0)
         
+        dalle = gr.Image(label="dalle", show_label=False)
         result = gr.Image(label="Result", show_label=False)
 
         # with gr.Accordion("Advanced Settings", open=False):
@@ -162,7 +166,7 @@ with gr.Blocks(css=css) as demo:
     run_button.click(
         fn = infer,
         inputs = [text, title],
-        outputs = [result]
+        outputs = [dalle,result]
     )
 
 demo.queue().launch()