Create app.py

app.py

@@ -0,0 +1,30 @@
+from huggingface_hub import from_pretrained_keras
+from keras_cv import models
+import gradio as gr
+
+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("lposti/dreambooth-piranesi")
+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
+    )
+    return generated_images 
+    
+output = gr.Gallery(label="Outputs").style(grid=(1,2))
+
+# customize interface
+title = "Dreambooth Piranesi"
+description = "This is a dreambooth model fine-tuned on the works of art of Giambattista Piranesi."
+examples=[["image of ancient city, in style sks", "image of futuristic city, in style sks"]]
+gr.Interface(infer, inputs=["text"], outputs=[output], title=title, description=description, examples=examples).queue().launch()