spaces

.gitignore

@@ -4,4 +4,5 @@ __pycache__/
 */__pycache__/
 models/
 plt/
-exp/
+exp/
+test_app.py

README.md

@@ -1,6 +1,6 @@
 ---
 title: DesignEdit
-emoji: ⚡
+emoji: 🌿🌼🌿
 colorFrom: pink
 colorTo: pink
 sdk: gradio

app.py

@@ -2,6 +2,7 @@ import os
 import subprocess
 import shlex
 from src.demo.model import DesignEdit
+import spaces
 
 os.makedirs('models', exist_ok=True)
 subprocess.run(shlex.split('wget https://huggingface.co/Adapter/DragonDiffusion/resolve/main/model/efficient_sam_vits.pt -O models/efficient_sam_vits.pt'))
@@ -52,6 +53,7 @@ with gr.Blocks(css='style.css') as demo:
             create_demo_layer(model.run_layer)
         with gr.TabItem('🔧 Mask Preparation: Draw or Sketch'):
             create_demo_mask_box(model.run_mask)
+
 demo.queue(concurrency_count=3, max_size=20)
 demo.launch(server_name="0.0.0.0")
 

requirements.txt

@@ -9,7 +9,7 @@ torch==2.0.1
 torchvision==0.15.2
 tqdm==4.65.0
 transformers==4.35.0
-gradio==3.50.2
+gradio==4.0.0
 basicsr==1.4.2
 accelerate==0.21.0
 invisible-watermark

src/demo/demo.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import numpy as np
 from src.demo.utils import get_point, store_img, get_point_move, store_img_move, clear_points, upload_image_move, segment_with_points, segment_with_points_paste, fun_clear, paste_with_mask_and_offset
+import spaces
+
 
 examples_remove = [
     [
@@ -165,6 +167,7 @@ examples_layer = [
 ]
 
 # 01
+@spaces.GPU
 def create_demo_remove(runner=None):
     DESCRIPTION = """
     # Object Removal
@@ -273,6 +276,7 @@ def create_demo_remove(runner=None):
 
 
 # 02:
+@spaces.GPU
 def create_demo_zooming(runner=None):
     DESCRIPTION = """
     # Zooming Out
@@ -342,6 +346,7 @@ def create_demo_zooming(runner=None):
         outputs=[original_image, width_scale, height_scale, output, new_image, new_mask])
     return demo
 # 03
+@spaces.GPU
 def create_demo_panning(runner=None):
     DESCRIPTION = """
     # Camera Panning
@@ -410,6 +415,7 @@ def create_demo_panning(runner=None):
         outputs=[original_image, w_direction, w_scale, h_direction, h_scale, new_image, new_mask, output])
     return demo
 # 04:
+@spaces.GPU
 def create_position_size(label=None):
     image = gr.Image(source='upload', label=label, interactive=True, type="numpy")
     with gr.Row():
@@ -442,6 +448,7 @@ def create_position_size(label=None):
         h_flip = gr.Radio(["down/up", "None"], value="None", label="Vertical Flip")
     return image, dx, dy, resize_scale, w_flip, h_flip
 # 05:
+@spaces.GPU
 def create_demo_layer(runner=None):
     DESCRIPTION = """
     # 🚩 Multi-Layered Editing 🚩
@@ -555,6 +562,7 @@ def create_demo_layer(runner=None):
     return demo
 
 # 06:
+@spaces.GPU
 def create_demo_mask_box(runner=None):
     DESCRIPTION = """
     # 🔧 Mask Preparation