Add title and project link to interface

app.py

@@ -8,6 +8,10 @@ DEFAULT_GEN_SIZE = 512
 
 def create_interface():
     with gr.Blocks() as app:
+        # Add main title and project link
+        gr.Markdown("# A simplified implementation of Inpaint4Drag")
+        gr.Markdown("## [Visit our project page for more examples](https://visual-ai.github.io/inpaint4drag)")
+        
         # State variables
         state = {
             'canvas_size': gr.Number(value=CANVAS_SIZE, visible=False, precision=0),

utils/drag.py

@@ -3,6 +3,8 @@ import cv2
 import torch
 from typing import Union
 
+import spaces
+
 def contour_to_points_and_mask(contour: np.ndarray, image_shape: tuple) -> tuple[np.ndarray, np.ndarray]:
     """Convert a contour to a set of points and binary mask.
     
@@ -187,6 +189,7 @@ def get_points_within_image_bounds(
 
     return valid_points_mask
 
+@spaces.GPU  
 def bi_warp(
     region_mask: np.ndarray,
     control_points: Union[np.ndarray, torch.Tensor],

utils/ui_utils.py

@@ -12,6 +12,7 @@ from diffusers import AutoPipelineForInpainting, AutoencoderTiny, LCMScheduler
 
 from utils.drag import bi_warp
 
+import spaces
 
 __all__ = [
     'clear_all', 'resize',
@@ -219,6 +220,7 @@ def preview_out_image(canvas, points, inpaint_ks):
         return image, None
 
 # Inpaint tools
+@spaces.GPU
 def setup_pipeline(device='cuda', model_version='v1-5'):
     """Initialize optimized inpainting pipeline with specified model configuration."""
     MODEL_CONFIGS = {
@@ -273,6 +275,7 @@ def get_pipeline():
             pipe.cached_prompt_embeds = pipe.encode_prompt('', 'cpu', 1, False)[0]
     return pipe
 
+@spaces.GPU
 def inpaint(image, inpaint_mask):
     """Perform efficient inpainting on masked regions using Stable Diffusion."""
     if image is None: