Update app.py

app.py

@@ -3,27 +3,23 @@ from PIL import Image
 import torch
 from diffusers import StableDiffusionUpscalePipeline
 
+# UltraSharpV2
 model_id = "Kim2091/UltraSharpV2"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe = StableDiffusionUpscalePipeline.from_pretrained(
-    model_id,
-    torch_dtype=torch.float16 if device == "cuda" else torch.float32
-)
+pipe = StableDiffusionUpscalePipeline.from_pretrained(model_id)
 pipe = pipe.to(device)
 
-def enhance_image(input_image: Image.Image) -> Image.Image:
-    result = pipe(input_image).images[0]
-    return result
+def enhance_image(image: Image.Image) -> Image.Image:
+    return pipe(image).images[0]
 
 demo = gr.Interface(
     fn=enhance_image,
     inputs=gr.Image(type="pil"),
     outputs=gr.Image(type="pil"),
-    title="UltraSharpV2 Image Enhancer",
-    description="Upload an image and UltraSharpV2 will enhance it.",
+    title="UltraSharpV2 Enhancer",
+    description="Enhances and upscales images (faces, objects, text).",
     allow_flagging="never"
 )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()