Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
-from gradio_client import Client, file
+from gradio_client import Client
 from PIL import Image
+import tempfile
 
 # Initialize the Hugging Face API clients
 captioning_client = Client("fancyfeast/joy-caption-pre-alpha")
@@ -8,10 +9,12 @@ generation_client = Client("black-forest-labs/FLUX.1-dev")
 
 # Function to caption an image
 def caption_image(image):
-    caption = captioning_client.predict(
-        input_image=image,
-        api_name="/stream_chat"
-    )
+    with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as temp_file:
+        image.save(temp_file.name)
+        caption = captioning_client.predict(
+            input_image=file(temp_file.name),
+            api_name="/stream_chat"
+        )
     return caption
 
 # Function to generate an image from a text prompt using Hugging Face API
@@ -45,7 +48,7 @@ def process_image(image, iterations):
         generated_images.append(new_image)
         
         # Set the newly generated image as the current image for the next iteration
-        current_image = new_image
+        current_image = Image.open(BytesIO(new_image))  # Convert to PIL Image
     
     return generated_images, captions