Update app.py

app.py

@@ -27,10 +27,13 @@ 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(device)
+upscaler_2 = None # UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
 text_to_image = None
 
+@spaces.GPU(duration=40)
 def load_model():
+    global upscaler_2
+    upscaler_2 = UpscaleWithModel.from_pretrained("Kim2091/ClearRealityV1").to(device)
     global text_to_image
     text_to_image = keras_hub.models.StableDiffusion3TextToImage.from_preset(
         "stable_diffusion_3_medium", width=768, height=768, dtype="bfloat16"