test

app.py

@@ -4,56 +4,23 @@ 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'))
-
 from src.demo.demo import *
 import shlex
 
 import cv2
 import gradio as gr
-pretrained_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
-model =  DesignEdit(pretrained_model_path=pretrained_model_path)
-DESCRIPTION_1 = """<div style="text-align: center; font-size: 80px;">
-        <strong class="title is-1">
-            <span style="color: green;">🌿D</span>
-            <span style="color: orange;">e</span>
-            <span style="color: rgb(63, 185, 63);">s</span>
-            <span style="color: green;">i</span>
-            <span style="color: rgb(200, 85, 23);">g</span>
-            <span style="color: green;">n</span>
-            <span style="color: orange;">E</span>
-            <span style="color: crimson;">d</span>
-            <span style="color: darkorange;">i</span>
-            <span style="color: green;">t🌿</span>
-          </strong> 
-    </div>
-    """
-DESCRIPTION_2 = """ <div style="text-align: center;font-size: 24px;"> <h1> Multi-Layered Latent Decomposition and Fusion for Unified & Accurate Image Editing</h1></div>"""
-DESCRIPTION_3 = """
-<div style="text-align: center; font-size: 24px;">
-    <p> Gradio demo for <a href="https://design-edit.github.io/">DesignEdit</a></p>
-</div>
-"""
 
-with gr.Blocks(css='style.css') as demo:
-    gr.Markdown(DESCRIPTION_1)
-    gr.Markdown(DESCRIPTION_2)
-    gr.Markdown(DESCRIPTION_3)
-    with gr.Tabs():
-        with gr.TabItem('1️⃣ Object Removal'):
-            create_demo_remove(model.run_remove)
-        with gr.TabItem('2️⃣ Zooming Out'):
-            create_demo_zooming(model.run_zooming)
-        with gr.TabItem('3️⃣ Camera Panning'):
-            create_demo_panning(model.run_panning)
-        with gr.TabItem('4️⃣ Object Moving, Resizing and Flipping'):
-            create_demo_moving(model.run_moving)
-        with gr.TabItem('5️⃣ 🚩 Multi-Layered Editing 🚩'):
-            create_demo_layer(model.run_layer)
-        with gr.TabItem('🔧 Mask Preparation: Draw or Sketch'):
-            create_demo_mask_box(model.run_mask)
+from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained(...)
+pipe.to('cuda')
 
-demo.queue(concurrency_count=3, max_size=20)
-demo.launch(server_name="0.0.0.0")
+@spaces.GPU
+def generate(prompt):
+    return pipe(prompt).images
 
+gr.Interface(
+    fn=generate,
+    inputs=gr.Text(),
+    outputs=gr.Gallery(),
+).launch()

app_old.py

@@ -0,0 +1,59 @@
+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'))
+
+from src.demo.demo import *
+import shlex
+
+import cv2
+import gradio as gr
+pretrained_model_path = "stabilityai/stable-diffusion-xl-base-1.0"
+model =  DesignEdit(pretrained_model_path=pretrained_model_path)
+DESCRIPTION_1 = """<div style="text-align: center; font-size: 80px;">
+        <strong class="title is-1">
+            <span style="color: green;">🌿D</span>
+            <span style="color: orange;">e</span>
+            <span style="color: rgb(63, 185, 63);">s</span>
+            <span style="color: green;">i</span>
+            <span style="color: rgb(200, 85, 23);">g</span>
+            <span style="color: green;">n</span>
+            <span style="color: orange;">E</span>
+            <span style="color: crimson;">d</span>
+            <span style="color: darkorange;">i</span>
+            <span style="color: green;">t🌿</span>
+          </strong> 
+    </div>
+    """
+DESCRIPTION_2 = """ <div style="text-align: center;font-size: 24px;"> <h1> Multi-Layered Latent Decomposition and Fusion for Unified & Accurate Image Editing</h1></div>"""
+DESCRIPTION_3 = """
+<div style="text-align: center; font-size: 24px;">
+    <p> Gradio demo for <a href="https://design-edit.github.io/">DesignEdit</a></p>
+</div>
+"""
+
+with gr.Blocks(css='style.css') as demo:
+    gr.Markdown(DESCRIPTION_1)
+    gr.Markdown(DESCRIPTION_2)
+    gr.Markdown(DESCRIPTION_3)
+    with gr.Tabs():
+        with gr.TabItem('1️⃣ Object Removal'):
+            create_demo_remove(model.run_remove)
+        with gr.TabItem('2️⃣ Zooming Out'):
+            create_demo_zooming(model.run_zooming)
+        with gr.TabItem('3️⃣ Camera Panning'):
+            create_demo_panning(model.run_panning)
+        with gr.TabItem('4️⃣ Object Moving, Resizing and Flipping'):
+            create_demo_moving(model.run_moving)
+        with gr.TabItem('5️⃣ 🚩 Multi-Layered Editing 🚩'):
+            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")
+