Update app.py

app.py

@@ -1,18 +1,15 @@
-# app.py
+# app.py (Versi Fix)
 
 import gradio as gr
-from realesrgan import RealESRGAN
-from PIL import Image
-import torch
+from super_image import EdsrModel, ImageLoader
 
-# Load model (langsung ke GPU jika tersedia)
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-model = RealESRGAN(device, scale=2)
-model.load_weights('weights/RealESRGAN_x2.pth')  # nanti download dari huggingface atau official
+# Load model dari Hugging Face
+model = EdsrModel.from_pretrained('eugenesiow/edsr-base', scale=2)
 
-def enhance_image(input_image):
-    output_image = model.predict(input_image)
+def enhance_image(image):
+    inputs = ImageLoader.load_image(image).unsqueeze(0)
+    preds = model(inputs)
+    output_image = ImageLoader.save_image(preds)
     return output_image
 
-# Launch API
-gr.Interface(fn=enhance_image, inputs=gr.Image(), outputs=gr.Image()).launch()
+gr.Interface(fn=enhance_image, inputs=gr.Image(type="pil"), outputs=gr.Image(type="pil")).launch()