Update app.py

app.py

@@ -3,6 +3,8 @@ from keras_cv import models
 import gradio as gr
 import tensorflow as tf 
 
+keras.mixed_precision.set_global_policy("mixed_float16")
+
 # load keras model
 resolution = 512
 dreambooth_model = models.StableDiffusion(