Hugging Face's logo

Spaces:
keras-dreambooth
/
dreambooth_diffusion_hokusai
Runtime error

App Files Community
1
vumichien commited on
Commit
e7ec95e
•
1 Parent(s): e299aee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
1
  from huggingface_hub import from_pretrained_keras
2
  from keras_cv import models
3
  import gradio as gr
4
- import tensorflow as tf
5
 
6
- keras.mixed_precision.set_global_policy("mixed_float16")
7
 
8
  # load keras model
9
  resolution = 512
 
1
  from huggingface_hub import from_pretrained_keras
2
  from keras_cv import models
3
  import gradio as gr
4
+ import tensorflow as tf
5
 
6
+ tf.keras.mixed_precision.set_global_policy("mixed_float16")
7
 
8
  # load keras model
9
  resolution = 512