Update app.py

app.py

@@ -1,19 +1,18 @@
 import gradio as gr
 from PIL import Image
-from image_gen_aux import UpscaleWithModel
-from image_gen_aux.utils import load_image
+from diffusers import StableDiffusionUpscalePipeline
+import torch
+
+# โหลดโมเดล
+model_id = "stabilityai/stable-diffusion-x4-upscaler"
+pipeline = StableDiffusionUpscalePipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipeline = pipeline.to("cuda")
 
 # ฟังก์ชันสำหรับการอัปสเกลภาพ
 def upscale_image(image):
-    # โหลดโมเดลจาก Hugging Face
-    model_selection = "4xNomosWebPhoto_esrgan"
-    original = load_image(image)
-
-    # ใช้โมเดลในการอัปเกรดภาพ
-    upscaler = UpscaleWithModel.from_pretrained(model_selection).to("cuda")
-    upscaled_image = upscaler(original, tiling=True, tile_width=1024, tile_height=1024)
-
-    return original, upscaled_image
+    # การอัปสเกลภาพ
+    upscaled_image = pipeline(image=image).images[0]
+    return upscaled_image
 
 # UI with Gradio
 with gr.Blocks() as demo: