Update app.py

app.py

@@ -3,8 +3,18 @@ import tensorflow as tf
 from huggingface_hub import from_pretrained_keras
 import numpy as np
 
+from huggingface_hub import hf_hub_download
+from keras.layers import TFSMLayer
+import numpy as np
+
+# Download the model
+model_path = hf_hub_download("keras-io/mobile-vit-xxs")
+
+# Load the model using TFSMLayer
+model = TFSMLayer(model_path, call_endpoint='serving_default')
 
-model = from_pretrained_keras("keras-io/mobile-vit-xxs")
+''' model = from_pretrained_keras("keras-io/mobile-vit-xxs") 
+'''
 
 classes=['dandelion','daisy','tulips','sunflower','rose']
 image_size = 256