Update app.py

app.py

@@ -8,18 +8,21 @@ API_KEY = os.getenv("HF_API_KEY")  # Fetch the API key from Hugging Face secrets
 
 headers = {"Authorization": f"Bearer {API_KEY}"}
 
-def generate_caption(image):
+def generate_caption(image_path):
     """Queries the Hugging Face API to generate an image caption."""
     if not API_KEY:
         return "API key is missing! Please set it in Hugging Face secrets."
     
     try:
-        response = requests.post(API_URL, headers=headers, files={"file": image})
-        response.raise_for_status()  # Raise an error for bad responses
-        result = response.json()
-        return result[0].get("generated_text", "No caption generated.")
+        with open(image_path, "rb") as image_file:
+            response = requests.post(API_URL, headers=headers, files={"file": image_file})
+            response.raise_for_status()  # Raise an error for bad responses
+            result = response.json()
+            return result[0].get("generated_text", "No caption generated.")
     except requests.exceptions.RequestException as e:
         return f"Error: {str(e)}"
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 # Gradio UI components
 title = "✨ Captionator_X ✨"
@@ -35,7 +38,7 @@ with gr.Blocks(theme=theme) as app:
     
     with gr.Row():
         with gr.Column():
-            image_input = gr.Image(type="file", label="Upload Your Image", elem_id="image-uploader")
+            image_input = gr.Image(type="filepath", label="Upload Your Image", elem_id="image-uploader")
         with gr.Column():
             output_text = gr.Textbox(label="Generated Caption", lines=4, interactive=False, elem_id="output-text", placeholder="Your caption will appear here...")