Update app.py

app.py

@@ -1,17 +1,17 @@
 import gradio as gr
-from transformers import pipeline
+from diffusion_single_file import load_model, enhance_image
 
-# Load the ZenCtrl tool pipeline for image enhancement
-enhancer = pipeline(model="fotographerai/zenctrl_tools")
+# Load the ZenCtrl model using the correct library
+model = load_model("fotographerai/zenctrl_tools")
 
-def enhance_image(input_image):
+def enhance_image_fn(input_image):
     # Enhance the image using the model
-    enhanced_image = enhancer(input_image)
+    enhanced_image = enhance_image(model, input_image)
     return enhanced_image
 
 # Set up the Gradio interface
 interface = gr.Interface(
-    fn=enhance_image,
+    fn=enhance_image_fn,
     inputs=gr.Image(type="pil", label="Upload Image to Enhance"),
     outputs=gr.Image(type="pil", label="Enhanced Image"),
     title="Image Enhancement with ZenCtrl",