Update app.py

app.py

@@ -1,7 +1,9 @@
 import gradio as gr
-from gradio_client import Client
+from gradio_client import Client, handle_file
 from PIL import Image
 import tempfile
+import requests
+from io import BytesIO
 
 # Initialize the Hugging Face API clients
 captioning_client = Client("fancyfeast/joy-caption-pre-alpha")
@@ -12,7 +14,7 @@ def caption_image(image):
     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),
+            input_image=handle_file(temp_file.name),
             api_name="/stream_chat"
         )
     return caption
@@ -29,7 +31,9 @@ def generate_image_from_caption(caption):
         num_inference_steps=28,
         api_name="/infer"
     )
-    return image
+    image_url = image[0]  # Assuming the API returns a URL to the image
+    response = requests.get(image_url)
+    return Image.open(BytesIO(response.content))
 
 # Main function to handle the upload and generate images and captions in a loop
 def process_image(image, iterations):
@@ -48,7 +52,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 = Image.open(BytesIO(new_image))  # Convert to PIL Image
+        current_image = new_image
     
     return generated_images, captions