Update app.py

app.py

@@ -2,16 +2,14 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 from PIL import Image
 import io
-import base64
 
 # Initialize the Inference Client
 client = InferenceClient()
 
 # Function to handle the image generation
-def generate_image(base64_image, prompt):
-    # Decode the base64 image
-    image_data = base64.b64decode(base64_image.split(",")[1])
-    image = Image.open(io.BytesIO(image_data))
+def generate_image(image, prompt):
+    # Open the uploaded image
+    image = Image.open(image)
 
     # Generate image using the InferenceClient
     generated_image = client.image_to_image(image, prompt=prompt)
@@ -22,16 +20,16 @@ def generate_image(base64_image, prompt):
     img_byte_arr.seek(0)  # Move the cursor to the beginning
 
     # Return the generated image
-    return img_byte_arr
+    return generated_image
 
 # Create Gradio interface
 iface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.inputs.Image(type="filepath", label="Input Image"),  # Filepath for uploaded image
+        gr.inputs.Image(type="file", label="Input Image"),  # File input for uploaded image
         gr.inputs.Textbox(label="Prompt")  # Textbox for the prompt
     ],
-    outputs=gr.outputs.Image(type="numpy", label="Generated Image"),  # Output as numpy array
+    outputs=gr.outputs.Image(type="pil", label="Generated Image"),  # Output as PIL image
     title="Image Generation with Hugging Face",
     description="Upload an image and provide a prompt to generate a new image."
 )