Fix the error

README.md

@@ -4,7 +4,7 @@ emoji: 📸
 colorFrom: green
 colorTo: pink
 sdk: gradio
-sdk_version: 3.39.0
+sdk_version: 4.44.1
 app_file: app.py
 pinned: false
 ---

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 
-from haystack.components.generators import HuggingFaceTGIGenerator
+from haystack.components.generators import HuggingFaceAPIGenerator
 from haystack.components.builders.prompt_builder import PromptBuilder
 from haystack import Pipeline
 from haystack.utils import Secret
@@ -35,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="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})
+    generator = HuggingFaceAPIGenerator(api_type="serverless_inference_api", api_params={"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)

requirements.txt

@@ -1,3 +1,3 @@
 haystack-ai
-gradio==3.39.0
+gradio
 transformers[torch,sentencepiece]