Update app.py

app.py

@@ -6,7 +6,18 @@ import os
 # Initialize the client
 client = Client("radames/Enhance-This-HiDiffusion-SDXL")
 
-def enhance_image(image_path, prompt):
+# Custom preprocessing function to save uploaded image to a temporary file
+def preprocess_image(image):
+    # Create a temporary directory
+    temp_dir = tempfile.mkdtemp()
+    # Save the image to a temporary file
+    image_path = os.path.join(temp_dir, "input_image.jpg")
+    image.save(image_path)
+    return image_path
+
+def enhance_image(image, prompt):
+    # Preprocess the image
+    image_path = preprocess_image(image)
     # Use the client to predict the result
     result = client.predict(image_path, prompt, api_name="/predict")
     return result
@@ -15,7 +26,7 @@ def enhance_image(image_path, prompt):
 iface = gr.Interface(
     fn=enhance_image,
     inputs=[
-        gr.Image(type="pil", label="Input Image", preprocess=file.preprocess_image, preprocess_kwargs={"image_mode": "RGB"}),
+        gr.Image(type="pil", label="Input Image"),
         gr.Textbox(lines=2, placeholder="Enter prompt here", label="Prompt")
     ],
     outputs=gr.Image(type="pil", label="Enhanced Image"),