update app

app.py

@@ -47,19 +47,19 @@ title = "KerasCV Stable Diffusion Demo on images of Ace."
 description = "This is a dreambooth model fine-tuned on images of my pug named Ace. To try it, input the concept with `puggieace dog`."
 examples=[
     ["A photo of puggieace dog in space", "", 10, 50],
-    ["A photo of a cute puggieace dog getting a haircut in a barbershop, ultra realistic, intricate details, highly detailed, photorealistic, octane render, 8 k, unreal engine. art by artgerm and greg rutkowski and charlie bowater and magali villeneuve and alphonse mucha", "", 12, 75],
-    ["Portrait of puggieace dog as a Roman Emperor, city in background, ultra realistic, intricate details, eerie, highly detailed, photorealistic, octane render, 8 k, unreal engine. art by artgerm and greg rutkowski and charlie bowater and magali villeneuve and alphonse mucha", "", 15, 75],
-    ["Photo of cute puggieace dog as an astronaut, space and planet in background, ultra realistic, concept art, intricate details, highly detailed, photorealistic, octane render, 8 k, unreal engine. trending on artstation", "", 15, 75],
-    ["Photo of cute puggieace dog as super hero, futuristic city in background, cinematic light, high dynamic range, insane intricate details, stunning cinema effects, aesthetic, masterpiece, trending on artstation, cartoon art", "", 12, 75],
+    # ["A photo of a cute puggieace dog getting a haircut in a barbershop, ultra realistic, intricate details, highly detailed, photorealistic, octane render, 8 k, unreal engine. art by artgerm and greg rutkowski and charlie bowater and magali villeneuve and alphonse mucha", "", 12, 75],
+    # ["Portrait of puggieace dog as a Roman Emperor, city in background, ultra realistic, intricate details, eerie, highly detailed, photorealistic, octane render, 8 k, unreal engine. art by artgerm and greg rutkowski and charlie bowater and magali villeneuve and alphonse mucha", "", 15, 75],
+    # ["Photo of cute puggieace dog as an astronaut, space and planet in background, ultra realistic, concept art, intricate details, highly detailed, photorealistic, octane render, 8 k, unreal engine. trending on artstation", "", 15, 75],
+    # ["Photo of cute puggieace dog as super hero, futuristic city in background, cinematic light, high dynamic range, insane intricate details, stunning cinema effects, aesthetic, masterpiece, trending on artstation, cartoon art", "", 12, 75],
 ]
 
 base_14 = "https://huggingface.co/nielsgl/dreambooth-pug-ace-sd1.4/resolve/main/"
 base_21 = "https://huggingface.co/nielsgl/dreambooth-keras-pug-ace-sd2.1/resolve/main/"
 
-model_card = f"""
+model_card_1 = f"""
 # KerasCV Stable Diffusion in Diffusers 🧨🤗
 
-DreamBooth model for the `puggieace` concept trained by nielsgl on the `nielsgl/dreambooth-ace` dataset.
+DreamBooth model for the `puggieace` concept trained by nielsgl on the [nielsgl/dreambooth-ace](https://huggingface.co/datasets/nielsgl/dreambooth-ace) dataset.
 It can be used by modifying the `instance_prompt`: **a photo of puggieace**.
 
 The examples are from 2 different Keras CV models (`StableDiffusion` and `StableDiffusionV2`, corresponding to Stable Diffusion V1.4 and V2.1, respectively) trained on the same dataset (`nielsgl/dreambooth-ace`).
@@ -69,6 +69,11 @@ The examples are from 2 different Keras CV models (`StableDiffusion` and `Stable
 The Stable Diffusion V2 pipeline contained in the corresponding repository (`nielsgl/dreambooth-keras-pug-ace-sd2.1`) was created using a modified version of [this Space](https://huggingface.co/spaces/sayakpaul/convert-kerascv-sd-diffusers) for StableDiffusionV2 from KerasCV. The purpose is to convert the KerasCV Stable Diffusion weights in a way that is compatible with [Diffusers](https://github.com/huggingface/diffusers). This allows users to fine-tune using KerasCV and use the fine-tuned weights in Diffusers taking advantage of its nifty features (like [schedulers](https://huggingface.co/docs/diffusers/main/en/using-diffusers/schedulers), [fast attention](https://huggingface.co/docs/diffusers/optimization/fp16), etc.).
 This model was created as part of the Keras DreamBooth Sprint 🔥. Visit the [organisation page](https://huggingface.co/keras-dreambooth) for instructions on how to take part!
 
+## Demo
+
+"""
+
+model_card_2 = f"""
 ## Examples
 
 ### Stable Diffusion V1.4
@@ -158,15 +163,43 @@ The following hyperparameters were used during training for Stable Diffusion v1.
 | training_precision | float32 |
 """
 
-with gr.Blocks() as card_interface:
-    gr.Markdown(model_card)
-
-demo_interface = gr.Interface(infer, inputs=[gr.Textbox(label="Positive Prompt", value="a photo of puggieace dog getting a haircut"),
-        gr.Textbox(label="Negative Prompt", value="bad anatomy, blurry"),
-        # gr.Slider(label='Number of gen image', minimum=1, maximum=4, value=2, step=1),
-        gr.Number(label='Guidance scale', value=12),
-        gr.Slider(label="Inference Steps",value=50),
-        ], outputs=[output], title=title, description=description, examples=examples).queue()
-
-
-gr.TabbedInterface([card_interface, demo_interface], ["Model Card", "Demo 🤗"]).launch()
+with gr.Blocks() as demo:
+    with gr.Row():
+        gr.Markdown(model_card_1)
+    with gr.Row():
+        # with gr.Column():
+        #     prompt_pos = gr.Textbox(label="Positive Prompt", value="a photo of puggieace dog getting a haircut")
+        #     prompt_neg = gr.Textbox(label="Negative Prompt", value="bad anatomy, blurry")
+        #     # gr.Slider(label='Number of gen image', minimum=1, maximum=4, value=2, step=1),
+        #     prompt_gs = gr.Number(label='Guidance scale', value=12)
+        #     prompt_steps = gr.Slider(label="Inference Steps",value=50)
+        #     prompt_btn = gr.Button("Generate")
+        #     prompt_btn.click(infer, inputs=[prompt_pos, prompt_neg, prompt_gs, prompt_steps])
+        gr.Interface(infer, inputs=[gr.Textbox(label="Positive Prompt", value="a photo of puggieace dog getting a haircut"),
+            gr.Textbox(label="Negative Prompt", value="bad anatomy, blurry"),
+            # gr.Slider(label='Number of gen image', minimum=1, maximum=4, value=2, step=1),
+            gr.Number(label='Guidance scale', value=12),
+            gr.Slider(label="Inference Steps",value=50),
+            ], outputs=[output], title=title, description=description, examples=examples).queue()
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown(model_card_2)
+        with gr.Column():
+            gr.Markdown(" ")
+
+demo.launch()
+
+
+
+# with gr.Blocks() as card_interface:
+#     gr.Markdown(model_card)
+
+# demo_interface = gr.Interface(infer, inputs=[gr.Textbox(label="Positive Prompt", value="a photo of puggieace dog getting a haircut"),
+#         gr.Textbox(label="Negative Prompt", value="bad anatomy, blurry"),
+#         # gr.Slider(label='Number of gen image', minimum=1, maximum=4, value=2, step=1),
+#         gr.Number(label='Guidance scale', value=12),
+#         gr.Slider(label="Inference Steps",value=50),
+#         ], outputs=[output], title=title, description=description, examples=examples).queue()
+
+
+# gr.TabbedInterface([card_interface, demo_interface], ["Model Card", "Demo 🤗"]).launch()