update

README.md

@@ -1,8 +1,8 @@
 ---
 title: Dreambooth Marvin Paranoid Android
-emoji: 🐢
-colorFrom: purple
-colorTo: blue
+emoji: 🤖
+colorFrom: white
+colorTo: black
 sdk: gradio
 sdk_version: 3.20.1
 app_file: app.py

app.py

@@ -13,39 +13,32 @@ loaded_diffusion_model = from_pretrained_keras("johko/dreambooth_marvin_paranoid
 dreambooth_model._diffusion_model = loaded_diffusion_model
 
 
-def generate_images(prompt: str, negative_prompt:str, num_imgs_to_gen: int, num_steps: int):
-    """
-    This function is used to generate images using our fine-tuned keras dreambooth stable diffusion model.
-    Args:
-        prompt (str): The text input given by the user based on which images will be generated.
-        num_imgs_to_gen (int): The number of images to be generated using given prompt.
-        num_steps (int): The number of denoising steps
-    Returns:
-        generated_img (List): List of images that were generated using the model
-    """
+def generate_images(prompt: str, negative_prompt: str, num_imgs_to_gen: int, num_steps: int, guidance_scale: float):
     generated_img = dreambooth_model.text_to_image(
         prompt, 
         negative_prompt=negative_prompt,
         batch_size=num_imgs_to_gen,
         num_steps=num_steps,
+        unconditional_guidance_scale=guidance_scale,
     )
-   
+
     return generated_img
-    
-with gr.Blocks() as demo:
-    gr.HTML("<h2 style=\"font-size: 2em; font-weight: bold\" align=\"center\">Keras Dreambooth - Marvin the Paranoid Android</h2>")    
-    with gr.Row():
-        with gr.Column():
-            prompt = gr.Textbox(lines=1, value="paranoid marvin a robot", label="Base Prompt")
-            negative_prompt = gr.Textbox(lines=1, value="deformed", label="Negative Prompt")
-            samples = gr.Slider(minimum=1, maximum=10, value=1, step=1, label="Number of Images")
-            num_steps = gr.Slider(label="Inference Steps",value=50)
-            run = gr.Button(value="Submit")
-        with gr.Column():
-            gallery = gr.Gallery(label="Outputs").style(grid=(1,2))
 
-    run.click(generate_images, inputs=[prompt,negative_prompt, samples, num_steps], outputs=gallery)
-    
-    gr.Examples([["A pillow looking like paranoid marvin a robot, 4k, high quality","deformed", 1, 50]],
-                [prompt,negative_prompt, samples,num_steps], gallery, generate_images)
-    gr.Markdown('\n Demo created by: <a href=\"https://huggingface.co/johko/\">johko</a>')
+
+# pass function, input type for prompt, the output for multiple images
+gr.Interface(
+    generate_images, 
+    [
+        gr.Textbox(label="Positive Prompt", value="a photo of paranoid marvin a robot"),
+        gr.Textbox(label="Negative Prompt", value="low quality, deformed"),
+        gr.Slider(label='Number of gen image', minimum=1, maximum=4, value=2, step=1),
+        gr.Slider(label="Inference Steps", value=50),
+        gr.Number(label='Guidance scale', value=7.5),
+    ], [
+        gr.Gallery(show_label=False).style(grid=(1,2)),
+    ],
+    title="Keras Dreambooth - Marvin the Paranoid Android",
+    description="This model has been fine-tuned to learn the concept of Marvin the paranoid Android from The Hitchhiker's Guide to the Galaxy.",
+    examples=[["A pillow looking like paranoid marvin a robot, 4k, high quality", "deformed", 1, 50]],
+    cache_examples=True
+    ).queue().launch(debug=True)