Update app.py

app.py

@@ -287,7 +287,8 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale):
     # ----- CUDA / precision / tiling -----
     use_cuda = False
     try:
-        use_cuda = hasattr(cv2, "cuda") and cv2.cuda.getCudaEnabledDeviceCount() > 0
+        import torch
+        use_cuda = torch.cuda.is_available()
     except Exception:
         use_cuda = False
     gpu_id = 0 if use_cuda else None
@@ -359,6 +360,7 @@ def render_progress(pct: float, text: str = "") -> str:
     label = f"<div style='font-size:12px;opacity:.8;margin-top:4px;'>{text} {pct:.1f}%</div>"
     return bar + label
 
+@GPU()
 def batch_realesrgan(
     files: list,             # from gr.Files (type='filepath')
     model_name: str,