Searching a GPU...

app.py

@@ -234,7 +234,8 @@ def uncrop(
             process_width = process_width - (process_width % 8)
             process_height = process_height - (process_height % 8)
 
-    progress(None, desc = "Processing...")
+    if torch.cuda.is_available():
+        progress(None, desc = "Searching a GPU...")
 
     output_image = uncrop_on_gpu(
         seed,
@@ -248,7 +249,8 @@ def uncrop(
         guidance_scale,
         image_guidance_scale,
         strength,
-        denoising_steps
+        denoising_steps,
+        progress
     )
 
     if limitation != "":
@@ -286,7 +288,9 @@ def uncrop_on_gpu(
         guidance_scale,
         image_guidance_scale,
         strength,
-        denoising_steps):
+        denoising_steps,
+        progress):
+    progress(None, desc = "Processing...")
     return pipe(
         seeds = [seed],
         width = process_width,