ZeroGPU

app.py

@@ -4,6 +4,7 @@ import time
 import math
 import random
 import torch
+import spaces
 
 from diffusers import StableDiffusionXLInpaintPipeline
 from PIL import Image, ImageFilter
@@ -232,21 +233,20 @@ def uncrop(
 
     progress(None, desc = "Processing...")
 
-    output_image = pipe(
-        seeds = [seed],
-        width = process_width,
-        height = process_height,
-        prompt = prompt,
-        negative_prompt = negative_prompt,
-        image = enlarged_image,
-        mask_image = mask_image,
-        num_inference_steps = num_inference_steps,
-        guidance_scale = guidance_scale,
-        image_guidance_scale = image_guidance_scale,
-        strength = strength,
-        denoising_steps = denoising_steps,
-        show_progress_bar = True
-    ).images[0]
+    output_image = uncrop_on_gpu(
+        seed,
+        process_width,
+        process_height,
+        prompt,
+        negative_prompt,
+        enlarged_image,
+        mask_image,
+        num_inference_steps,
+        guidance_scale,
+        image_guidance_scale,
+        strength,
+        denoising_steps
+    )
 
     if limitation != "":
         output_image = output_image.resize((output_width, output_height))
@@ -270,6 +270,36 @@ def uncrop(
         mask_image
     ]
 
+@spaces.GPU(duration=420)
+def uncrop_on_gpu(
+        seed,
+        process_width,
+        process_height,
+        prompt,
+        negative_prompt,
+        enlarged_image,
+        mask_image,
+        num_inference_steps,
+        guidance_scale,
+        image_guidance_scale,
+        strength,
+        denoising_steps):
+    return pipe(
+        seeds = [seed],
+        width = process_width,
+        height = process_height,
+        prompt = prompt,
+        negative_prompt = negative_prompt,
+        image = enlarged_image,
+        mask_image = mask_image,
+        num_inference_steps = num_inference_steps,
+        guidance_scale = guidance_scale,
+        image_guidance_scale = image_guidance_scale,
+        strength = strength,
+        denoising_steps = denoising_steps,
+        show_progress_bar = True
+    ).images[0]
+
 with gr.Blocks() as interface:
     gr.HTML(
         """
@@ -289,7 +319,8 @@ with gr.Blocks() as interface:
         <li>To modify <b>anything else</b> on your image, I recommend to use <i>Instruct Pix2Pix</i>.</li>
         </ul>
         <br/>
-        🐌 Slow process... ~1 hour. Your computer must <u>not</u> enter into standby mode.<br/>You can duplicate this space on a free account, it works on CPU and should also run on CUDA.<br/>
+        """ + ("🏃‍♀️ Estimated time: few minutes." if torch.cuda.is_available() else "🐌 Slow process... ~1 hour.") + """
+        Your computer must <u>not</u> enter into standby mode.<br/>You can duplicate this space on a free account, it's designed to work on CPU, GPU and ZeroGPU.<br/>
         <a href='https://huggingface.co/spaces/Fabrice-TIERCELIN/Uncrop?duplicate=true'><img src='https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14'></a>
         <br/>
         ⚖️ You can use, modify and share the generated images but not for commercial uses.