Depth2Img Mirage

app.py

@@ -11,6 +11,13 @@ dept2img = StableDiffusionDepth2ImgPipeline.from_pretrained(
     torch_dtype=torch.float16,
 ).to(device)
 
+css = '''
+    .instruction{position: absolute; top: 0;right: 0;margin-top: 0px !important}
+    .arrow{position: absolute;top: 0;right: -110px;margin-top: -8px !important}
+    #component-4, #component-3, #component-10{min-height: 0}
+    .duplicate-button img{margin: 0}
+'''
+
 
 def pad_image(input_image):
     pad_w, pad_h = np.max(((2, 2), np.ceil(
@@ -60,15 +67,30 @@ block = gr.Blocks().queue()
 with block:
     with gr.Row():
         with gr.Column():
-            gr.Markdown("## Stable Diffusion 2 Depth2Img")
-            gr.HTML("<p><a href='https://huggingface.co/spaces/radames/stable-diffusion-depth2img?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' alt='Duplicate Space'></a></p>")
+            top_description = gr.HTML(f'''
+                    <div style="text-align: center; max-width: 650px; margin: 0 auto;">
+                    <div>
+                        <img class="logo" src="file/mirage.png" alt="Mirage Logo"
+                            style="margin: auto; max-width: 7rem;">
+                        <br />
+                        <h1 style="font-weight: 900; font-size: 2.5rem;">
+                        Depth2Img Web UI
+                        </h1>
+                        <br />
+                        <a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/MirageML/dreambooth?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" alt="Duplicate Space"></a>
+                    </div>
+                    </div>
+                ''')
+            # gr.Markdown("## Stable Diffusion 2 Depth2Img")
+            # gr.HTML("<p><a href='https://huggingface.co/spaces/radames/stable-diffusion-depth2img?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' alt='Duplicate Space'></a></p>")
 
 
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(source='upload', type="pil")
-            depth_image = gr.Image(
-                source='upload', type="pil", label="Depth image Optional", value=None)
+            # depth_image = gr.Image(
+            #     source='upload', type="pil", label="Depth image Optional", value=None)
+            depth_image = None
             prompt = gr.Textbox(label="Prompt")
             negative_prompt = gr.Textbox(label="Negative Pompt")
 
@@ -77,7 +99,7 @@ with block:
                 num_samples = gr.Slider(
                     label="Images", minimum=1, maximum=4, value=1, step=1)
                 steps = gr.Slider(label="Steps", minimum=1,
-                                  maximum=50, value=50, step=1)
+                                  maximum=100, value=50, step=1)
                 scale = gr.Slider(
                     label="Guidance Scale", minimum=0.1, maximum=30.0, value=9.0, step=0.1
                 )
@@ -96,22 +118,20 @@ with block:
                 grid=[2], height="auto")
     gr.Examples(
         examples=[
-            ["./examples/baby.jpg", "high definition photo of a baby astronaut space walking at the international space station with earth seeing from above in the background",
-             "", 50, 4, 9.0, 123123123, 0.8, None],
-            ["./examples/gol.jpg", "professional photo of a Elmo jumping between two high rises, beautiful colorful city landscape in the background",
-             "", 50, 4, 9.0, 1734133747, 0.9, None],
-            ["./examples/bag.jpg", "a photo of a bag of cookies in the bathroom", "low light, dark, blurry", 50, 4, 9.0, 1734133747, 0.9, "./examples/depth.jpg"],
-            ["./examples/smile_face.jpg", "a hand holding a very spherical orange", "low light, dark, blurry", 50, 4, 6.0, 961736534, 0.5, "./examples/smile_depth.jpg"]
+            ["./examples/original_iso.png", "hogwarts castle",
+             "", 50, 4, 10.0, 123123123, 0.8],
+            ["./examples/original_sword.png", "flaming sword",
+             "", 50, 4, 9.0, 1734133747, 0.8],
 
         ],
         inputs=[input_image, prompt, negative_prompt, steps,
-                num_samples, scale, seed, strength, depth_image],
+                num_samples, scale, seed, strength],
         outputs=[gallery],
         fn=predict,
         cache_examples=True,
     )
     run_button.click(fn=predict, inputs=[input_image, prompt, negative_prompt,
-                     steps, num_samples, scale, seed, strength, depth_image], outputs=[gallery])
+                     steps, num_samples, scale, seed, strength], outputs=[gallery])
 
 
 block.launch(show_api=False)