Update app.py

app.py

@@ -1,27 +1,13 @@
 import gradio as gr
 from gradio_client import Client
-import base64
-from PIL import Image
-import io
 
 # Initialize the client
 client = Client("radames/Enhance-This-HiDiffusion-SDXL")
 
-def pil_to_base64(img):
-    buffered = io.BytesIO()
-    img.save(buffered, format="PNG")
-    return base64.b64encode(buffered.getvalue()).decode('utf-8')
-
 def enhance_image(image, prompt):
-    # Convert the image to base64
-    image_base64 = pil_to_base64(image)
-    
     # Use the client to predict the result
-    result_base64 = client.predict(image_base64, prompt, api_name="/predict")
-    
-    # Convert result from base64 to PIL Image
-    result_image = Image.open(io.BytesIO(base64.b64decode(result_base64)))
-    return result_image
+    result = client.predict(image, prompt, api_name="/predict")
+    return result
 
 # Create the Gradio interface
 iface = gr.Interface(