ksvmuralidhar commited on
Commit
7301ba2
1 Parent(s): e02b592

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -32,6 +32,7 @@ class ImageVectorizer:
32
  self.__model = self.get_model()
33
 
34
  @staticmethod
 
35
  def get_model():
36
  model = load_model('vegetable_classification_model_vgg.h5') # loading saved VGG model finetuned on vegetable images for classification
37
  top = model.get_layer('block5_pool').output
 
32
  self.__model = self.get_model()
33
 
34
  @staticmethod
35
+ @st.cache_resource
36
  def get_model():
37
  model = load_model('vegetable_classification_model_vgg.h5') # loading saved VGG model finetuned on vegetable images for classification
38
  top = model.get_layer('block5_pool').output