app.py

# app.py (Versi Fix)

import gradio as gr
from super_image import EdsrModel, ImageLoader

# Load model dari Hugging Face
model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2)

def enhance_image(image):
    inputs = ImageLoader.load_image(image).unsqueeze(0)
    preds = model(inputs)
    output_image = ImageLoader.save_image(preds)
    return output_image

gr.Interface(fn=enhance_image, inputs=gr.Image(type="pil"), outputs=gr.Image(type="pil")).launch()