Spaces:
Sleeping
Sleeping
ksvmuralidhar
commited on
Commit
•
7301ba2
1
Parent(s):
e02b592
Update app.py
Browse files
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
|