Update app.py

app.py

@@ -15,6 +15,8 @@ import keras
 import keras_hub
 import torch
 
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
 torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
@@ -25,7 +27,7 @@ torch.backends.cudnn.benchmark = False
 #torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
 
-upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(torch.device('cuda'))
+upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
 
 text_to_image = keras_hub.models.StableDiffusion3TextToImage.from_preset(
     "stable_diffusion_3_medium", width=768, height=768, dtype="bfloat16"