Update app.py

app.py

@@ -3,7 +3,6 @@ import os
 import io
 from PIL import Image
 import requests
-import json
 import warnings
 import gradio as gr
 
@@ -31,11 +30,14 @@ def generate_caption(image):
     try:
         headers = {"Authorization": f"Bearer {hf_api_key}"}
         
-        # Convert PIL image to binary stream
+        # Convert the PIL image to a binary stream in JPEG format
         buffered = io.BytesIO()
+        image = image.convert("RGB")  # Ensure the image is in RGB mode
         image.save(buffered, format="JPEG")
         buffered.seek(0)
-        files = {"inputs": ("image.jpg", buffered, "image/jpeg")}
+        
+        # Create the appropriate payload for the API
+        files = {"file": ("image.jpg", buffered, "image/jpeg")}
         
         # Make the POST request to the endpoint
         response = requests.post(endpoint_url, headers=headers, files=files)