up

app.py

@@ -40,6 +40,8 @@ class Model:
 
 models = [
     Model("Stable Diffusion v1-4", "CompVis/stable-diffusion-v1-4"),
+    # Model("Stable Diffusion v1-5", "runwayml/stable-diffusion-v1-5"),
+    Model("anything-v4.0", "andite/anything-v4.0"),
 ]
 
 MODELS = {m.name: m for m in models}
@@ -140,10 +142,10 @@ with gr.Blocks(css="style.css") as demo:
         f"""
             <div class="finetuned-diffusion-div">
               <div>
-                <h1>Protogen Diffusion</h1>
+                <h1>Stable Diffusion Latent Upscaler</h1>
               </div>
               <p>
-               Demo for multiple fine-tuned Protogen Stable Diffusion models.
+               Demo for Latent Diffusion Upscaling
               </p>
               <p>
                Running on <b>{device}</b>
@@ -162,15 +164,6 @@ with gr.Blocks(css="style.css") as demo:
                     choices=[m.name for m in models],
                     value=models[0].name,
                 )
-                with gr.Box(visible=False) as custom_model_group:
-                    custom_model_path = gr.Textbox(
-                        label="Custom model path",
-                        placeholder="Path to model, e.g. darkstorm2150/Protogen_x3.4_Official_Release",
-                        interactive=True,
-                    )
-                    gr.HTML(
-                        "<div><font size='2'>Custom models have to be downloaded first, so give it some time.</font></div>"
-                    )
 
                 with gr.Row():
                     prompt = gr.Textbox(
@@ -183,14 +176,13 @@ with gr.Blocks(css="style.css") as demo:
                         rounded=(False, True, True, False)
                     )
 
-                # image_out = gr.Image(height=512)
                 low_res_image = gr.Gallery(
                     label="512-pix image", show_label=False, elem_id="gallery"
-                ).style(grid=[2], height="auto")
+                ).style(grid=[2], height=512)
 
                 up_res_image = gr.Gallery(
                     label="1024-pix image", show_label=False, elem_id="gallery"
-                ).style(grid=[2], height="auto")
+                ).style(grid=[2], height=1024)
 
             state_info = gr.Textbox(label="State", show_label=False, max_lines=2).style(
                 container=False
@@ -235,9 +227,9 @@ with gr.Blocks(css="style.css") as demo:
 
 #    ex = gr.Examples(
 #        [
-#            [models[0].name, "portrait of a beautiful alyx vance half life", 10, 50, "canvas frame, ((disfigured)), ((bad art)), ((deformed)),((extra limbs)),((close up)),((b&w)), weird colors, blurry, (((duplicate))), ((morbid)), ((mutilated)), [out of frame], extra fingers, mutated hands, ((poorly drawn hands)), ((poorly drawn face)), (((mutation))), (((deformed))), blurry, ((bad anatomy)), (((bad proportions))), ((extra limbs)), cloned face, (((disfigured))), out of frame, ugly, extra limbs, (bad anatomy), gross proportions, (malformed limbs), ((missing arms)), ((missing legs)), (((extra arms))), (((extra legs))), mutated hands, (fused fingers), (too many fingers), (((long neck))), Photoshop, video game, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, mutation, mutated, extra limbs, extra legs, extra arms, disfigured, deformed, cross-eye, body out of frame, blurry, bad art, bad anatomy"],
+#            [models[0].name, "a photo of an astronaut high resolution, unreal engine, ultra realistic", 7.5, 50, 33, ""]
 #        ],
-#        inputs=[model_name, prompt, guidance, steps, neg_prompt],
+#        inputs=[model_name, prompt, guidance, steps, seed, neg_prompt],
 #        outputs=outputs,
 #        fn=inference,
 #        cache_examples=False,
@@ -247,10 +239,10 @@ with gr.Blocks(css="style.css") as demo:
         """
     <div style="border-top: 1px solid #303030;">
       <br>
-      <p>Models by <a href="https://huggingface.co/darkstorm2150">@darkstorm2150</a> and others. ❤️</p>
+      <p>Models by 🤗 Hugging Face and others. ❤️</p>
       <p>This space uses the <a href="https://github.com/LuChengTHU/dpm-solver">DPM-Solver++</a> sampler by <a href="https://arxiv.org/abs/2206.00927">Cheng Lu, et al.</a>.</p>
-      <p>Space by: Darkstorm (Victor Espinoza)<br>
-      <a href="https://www.instagram.com/officialvictorespinoza/">Instagram</a>
+      <p>This is a Demo Space For:<br>
+      <a href="https://huggingface.co/stabilityai/sd-x2-latent-upscaler">Stability AI's Latent Upscaler</a>
     </div>
     """
     )