initialize zerogpu

app.py

@@ -6,7 +6,7 @@ from diffusers.pipelines.auto_pipeline import AutoPipelineForImage2Image
 from src.sdxl_inversion_pipeline import SDXLDDIMPipeline
 from src.config import RunConfig
 from src.editor import ImageEditorDemo
-
+import spaces
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 scheduler_class = MyEulerAncestralDiscreteScheduler
@@ -27,7 +27,7 @@ pipe_inversion.scheduler_inference = scheduler_class.from_config(pipe_inference.
 #     pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
 #     pipe = pipe.to(device)
 
-
+@spaces.GPU
 def infer(input_image, description_prompt, target_prompt, edit_guidance_scale, num_inference_steps=4,
           num_inversion_steps=4,
           inversion_max_step=0.6):
@@ -65,8 +65,8 @@ else:
     power_device = "CPU"
 
 # with gr.Blocks(css=css) as demo:
-with gr.Blocks() as demo:
-    gr.Markdown(f"""
+with gr.Blocks(css="style.css") as demo:
+    gr.Markdown(f""" # Real Time Editing with RNRI Inversion 🍎⚡️
     This is a demo for our [paper](https://arxiv.org/abs/2312.12540) **RNRI: Regularized Newton Raphson Inversion for Text-to-Image Diffusion Models**.
     Image editing using our RNRI for inversion demonstrates significant speed-up and improved quality compared to previous state-of-the-art methods.
     Take a look at our [project page](https://barakmam.github.io/rnri.github.io/).
@@ -79,18 +79,20 @@ with gr.Blocks() as demo:
             with gr.Row():
                 description_prompt = gr.Text(
                     label="Image description",
+                    info = "Enter your image description ",
                     show_label=False,
                     max_lines=1,
-                    placeholder="Enter your image description",
+                    placeholder="a cake on a table",
                     container=False,
                 )
 
             with gr.Row():
                 target_prompt = gr.Text(
                     label="Edit prompt",
+                    info = "Enter your edit prompt",
                     show_label=False,
                     max_lines=1,
-                    placeholder="Enter your edit prompt",
+                    placeholder="an oreo cake on a table",
                     container=False,
                 )
 

style.css

@@ -0,0 +1,16 @@
+
+#component-0{
+  max-width: 900px;
+  margin: 0 auto;
+}
+
+#description, h1 {
+  text-align: center;
+}
+
+#duplicate-button {
+  margin: auto;
+  color: #fff;
+  background: #1565c0;
+  border-radius: 100vh;
+}