Update app.py

app.py

@@ -1,18 +1,20 @@
 import torch
 import gradio as gr
 from PIL import Image
-from transformers import AutoImageProcessor, DeepLPFModel
+from transformers import AutoImageProcessor, AutoModelForImageProcessing
 
 # Load model and processor
-processor = AutoImageProcessor.from_pretrained("michaelbenayoun/deeplpf")
-model = DeepLPFModel.from_pretrained("michaelbenayoun/deeplpf").to('cpu')
+processor = AutoImageProcessor.from_pretrained("eugenesiow/EnhanceNet")
+model = AutoModelForImageProcessing.from_pretrained("eugenesiow/EnhanceNet").to('cpu')
 
 def enhance_image(image):
     # Convert PIL Image to tensor
     inputs = processor(images=image, return_tensors="pt").to('cpu')
+    
     # Forward pass
     with torch.no_grad():
         outputs = model(**inputs)
+    
     # Convert tensor back to PIL Image
     enhanced_image = processor.post_process_output(outputs.pixel_values)
     return enhanced_image[0]
@@ -23,7 +25,7 @@ iface = gr.Interface(
     inputs=gr.Image(type="pil", label="Upload Dull Image"),
     outputs=gr.Image(type="pil", label="Vibrant Image"),
     title="Dull to Vibrant Image Enhancer",
-    description="Upload a dull image to enhance its vibrancy using DeepLPF on CPU."
+    description="Enhance dull images using EnhanceNet on CPU. For faster GPU processing, duplicate this Space and enable GPU in settings."
 )
 
 if __name__ == "__main__":