Update app.py

app.py

@@ -5,7 +5,6 @@ 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 = """
@@ -16,7 +15,7 @@ description = """
 * Choose your model
 * ✨ Captionate! ✨ 
 
-It uses [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base) model for image2task caption generation task.
+It uses [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base) model for image-to-text 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.
 
@@ -34,11 +33,10 @@ Instagram Caption:
 hf_api_key = os.environ["HF_API_KEY"]
 
 def generate_caption(image_file_path, model_name):
-    image_to_text = ImageCaptioner(
-        model_name="Salesforce/blip-image-captioning-base",
-    )
+    image_to_text = ImageCaptioner(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":100})
+    
     captioning_pipeline = Pipeline()
     captioning_pipeline.add_component("image_to_text", image_to_text)
     captioning_pipeline.add_component("prompt_builder", prompt_builder)
@@ -55,7 +53,11 @@ with gr.Blocks(theme="soft") as demo:
     with gr.Row():
         image = gr.Image(type="filepath")
         with gr.Column():
-            model_name = gr.Dropdown(["mistralai/Mistral-7B-Instruct-v0.2","OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", "tiiuae/falcon-7b-instruct", "tiiuae/falcon-7b", "HuggingFaceH4/starchat-beta", "bigscience/bloom", "google/flan-t5-xxl"], value="mistralai/Mistral-7B-Instruct-v0.2", label="Choose your model!")
+            model_name = gr.Dropdown(
+                ["mistralai/Mistral-7B-Instruct-v0.2","OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", "tiiuae/falcon-7b-instruct", "tiiuae/falcon-7b", "HuggingFaceH4/starchat-beta", "bigscience/bloom", "google/flan-t5-xxl"], 
+                value="mistralai/Mistral-7B-Instruct-v0.2", 
+                label="Choose your model!"
+                )
             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)