More examples

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text
+*.jpeg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -6,10 +6,14 @@ from haystack.nodes import PromptNode, PromptTemplate
 from haystack import Pipeline
 
 description = """
-# Captionate ✨ 📸
-### Create Instagram captions for your pics! 
+# Captionate 📸
+### Create Instagram captions for your pics!
 
-`OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5` and `tiiuae/falcon-7b-instruct` perform the best but try out different model to see how they react to the same prompt.
+* Upload your photo or select one from examples
+* Choose your model
+* ✨ Captionate! ✨ 
+
+`OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5` and `tiiuae/falcon-7b-instruct` perform the best but try out different models to see how they react to the same prompt.
 
 Built by [Bilge Yucel](https://twitter.com/bilgeycl) using [Haystack](https://github.com/deepset-ai/haystack) 💙
 """
@@ -27,7 +31,7 @@ Descriptive text: {documents};
 Instagram Caption:
 """)
 
-hf_api_key = "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"
+hf_api_key = os.environ["HF_API_KEY"]
 
 def generate_caption(image_file_paths, model_name):
     captioning_pipeline = Pipeline()
@@ -37,13 +41,13 @@ def generate_caption(image_file_paths, model_name):
     caption = captioning_pipeline.run(file_paths=[image_file_paths])
     return caption["results"][0]
 
-with gr.Blocks(theme="monochrome") as demo:
+with gr.Blocks(theme="soft") as demo:
     gr.Markdown(value=description)
     with gr.Row():
         image = gr.Image(type="filepath")
         with gr.Column():
             model_name = gr.Dropdown(["OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", "tiiuae/falcon-7b-instruct", "tiiuae/falcon-7b", "EleutherAI/gpt-neox-20b", "HuggingFaceH4/starchat-beta", "bigscience/bloom"], value="OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", label="Choose your model!")
-            gr.Examples(["./whale.png"], examples_per_page=1, inputs=image, label="Click on any example") 
+            gr.Examples(["./whale.png", "./rainbow.jpeg", "./selfie.png"], inputs=image, label="Click on any example") 
     submit_btn = gr.Button("✨ Captionate ✨")
     caption = gr.Textbox(label="Caption", show_copy_button=True)
     submit_btn.click(fn=generate_caption, inputs=[image, model_name], outputs=[caption])