Update app.py

app.py

@@ -5,6 +5,7 @@ from haystack.components.generators import HuggingFaceTGIGenerator
 from haystack.components.builders.prompt_builder import PromptBuilder
 from haystack import Pipeline
 from haystack.utils import Secret
+
 from image_captioner import ImageCaptioner
 
 description = """
@@ -15,7 +16,9 @@ description = """
 * Choose your model
 * ✨ Captionate! ✨ 
 
-`mistralai/Mistral-7B-Instruct-v0.2` performs best, but try different models to see how they react to the same prompt.
+It uses [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base) model for image2task caption generation task.
+
+For Instagrammable captions, `mistralai/Mistral-7B-Instruct-v0.2` performs best, but try different models to see how they react to the same prompt.
 
 Built by [Bilge Yucel](https://twitter.com/bilgeycl) using [Haystack 2.0](https://github.com/deepset-ai/haystack) 💙
 """
@@ -32,7 +35,7 @@ hf_api_key = os.environ["HF_API_KEY"]
 
 def generate_caption(image_file_path, model_name):
     image_to_text = ImageCaptioner(
-        model_name="nlpconnect/vit-gpt2-image-captioning",
+        model_name="Salesforce/blip-image-captioning-base",
     )
     prompt_builder = PromptBuilder(template=prompt_template)
     generator = HuggingFaceTGIGenerator(model=model_name, token=Secret.from_token(hf_api_key), generation_kwargs={"max_new_tokens":50})