Update app.py

app.py

@@ -1,6 +1,7 @@
 from huggingface_hub import from_pretrained_keras
 import keras_cv
 import gradio as gr
+from tensorflow import keras
 
 keras.mixed_precision.set_global_policy("mixed_float16")
 # load keras model