Update app.py

app.py

@@ -1,22 +1,24 @@
 import gradio as gr
-from gradio_client import Client
+from gradio_client import Client, file
+import tempfile
+import os
 
 # Initialize the client
 client = Client("radames/Enhance-This-HiDiffusion-SDXL")
 
-def enhance_image(image, prompt):
+def enhance_image(image_path, prompt):
     # Use the client to predict the result
-    result = client.predict(image, prompt, api_name="/predict")
+    result = client.predict(image_path, prompt, api_name="/predict")
     return result
 
 # Create the Gradio interface
 iface = gr.Interface(
     fn=enhance_image,
     inputs=[
-        gr.Image(type="pil", label="Input Image"),
-        gr.Textbox(lines=2, placeholder="Enter prompt here", label="Prompt")
+        gr.inputs.Image(type="pil", label="Input Image", preprocess=file.preprocess_image, preprocess_kwargs={"image_mode": "RGB"}),
+        gr.inputs.Textbox(lines=2, placeholder="Enter prompt here", label="Prompt")
     ],
-    outputs=gr.Image(type="pil", label="Enhanced Image"),
+    outputs=gr.outputs.Image(type="pil", label="Enhanced Image"),
     title="Image Enhancement with Text Prompt",
     description="Upload an image and provide a text prompt to enhance the image using HiDiffusion SDXL."
 )