Update the ui

app.py

@@ -7,9 +7,11 @@ from haystack import Pipeline
 
 description = """
 # Captionate ✨ 📸
-## Create Instagram captions for your insta pics! 
+## Create Instagram captions for your pics! 
 
-Built by [Bilge Yucel](https://twitter.com/bilgeycl) using [Haystack](https://github.com/deepset-ai/haystack)💙
+
+
+Built by [Bilge Yucel](https://twitter.com/bilgeycl) using [Haystack](https://github.com/deepset-ai/haystack) 💙
 """
 
 image_to_text = TransformersImageToText(
@@ -19,9 +21,9 @@ image_to_text = TransformersImageToText(
 
 prompt_template = PromptTemplate(prompt="""
 You will receive a descriptive text of a photo.
-Try to come up with a nice Instagram caption that has a phrase rhyming with the text.
+Try to come up with a nice Instagram caption that has a phrase rhyming with the text. Include emojis to the caption.
                                                                                                                                    
-Descriptive text:{documents};
+Descriptive text: {documents};
 Caption:
 """)
 
@@ -33,15 +35,15 @@ def generate_caption(image_file_paths, model_name):
     captioning_pipeline.add_node(component=image_to_text, name="image_to_text", inputs=["File"])
     captioning_pipeline.add_node(component=prompt_node, name="prompt_node", inputs=["image_to_text"])
     caption = captioning_pipeline.run(file_paths=[image_file_paths])
-    print(caption)
     return caption["results"][0]
 
-with gr.Blocks(theme="soft") as demo:
+with gr.Blocks(theme="monochrome") as demo:
     gr.Markdown(value=description)
-    image = gr.Image(type="filepath")
-    model_name = gr.Dropdown(["tiiuae/falcon-7b-instruct", "tiiuae/falcon-7b", "EleutherAI/gpt-neox-20b", "HuggingFaceH4/starchat-beta", "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", "timdettmers/guanaco-33b-merged", "bigscience/bloom"], value="tiiuae/falcon-7b-instruct", label="Choose your model!")
+    with gr.Row():
+        image = gr.Image(type="filepath")
+        model_name = gr.Dropdown(["tiiuae/falcon-7b-instruct", "tiiuae/falcon-7b", "EleutherAI/gpt-neox-20b", "HuggingFaceH4/starchat-beta", "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", "bigscience/bloom"], value="tiiuae/falcon-7b-instruct", label="Choose your model!")
     submit_btn = gr.Button("✨ Captionate ✨")
-    caption = gr.Textbox(label="Caption")
+    caption = gr.Textbox(label="Caption", show_copy_button=True)
     submit_btn.click(fn=generate_caption, inputs=[image, model_name], outputs=[caption])
 
 if __name__ == "__main__":