Update app.py

app.py

@@ -1,25 +1,30 @@
-from utils_app import keras_stable_diffusion_app
+from huggingface_hub import from_pretrained_keras
+from keras_cv import models
 import gradio as gr
 
-app = gr.Blocks()
-with app:
-    gr.HTML(
-        """
-        <h1 style='text-align: center'>
-        Keras Diffusion WebUI
-        </h1>
-        """
-    )
-    gr.Markdown(
-        """
-        <h4 style='text-align: center'>
-        Follow me for more! 
-        <a href='https://twitter.com/kadirnar_ai' target='_blank'>Twitter</a> | <a href='https://github.com/kadirnar' target='_blank'>Github</a> | <a href='https://www.linkedin.com/in/kadir-nar/' target='_blank'>Linkedin</a>
-        </h4>
-        """
+from tensorflow import keras
+
+keras.mixed_precision.set_global_policy("mixed_float16")
+
+# prepare model
+resolution = 512
+sd_dreambooth_model = models.StableDiffusion(
+    img_width=resolution, img_height=resolution
+)
+db_diffusion_model = from_pretrained_keras("kadirnar/dreambooth_diffusion_model_v5")
+sd_dreambooth_model._diffusion_model = db_diffusion_model
+
+# generate images
+def infer(prompt):
+    generated_images = sd_dreambooth_model.text_to_image(
+        prompt, batch_size=2
     )
-    with gr.Row():
-        with gr.Column():
-            keras_diffusion_app = keras_stable_diffusion_app()
+    return generated_images 
+    
+output = gr.Gallery(label="Outputs").style(grid=(1,2))
 
-app.queue().launch(debug=True)  
+# customize interface
+title = "Dreambooth Demo on Traditional Furniture Images"
+description = "This is a dreambooth model fine-tuned on dog images. To try it, input the concept with {sks dog}."
+examples=[["sks traditional furniture"]]
+gr.Interface(infer, inputs=["text"], outputs=[output], title=title, description=description, examples=examples).queue().launch()