tonyliu404 commited on
Commit
7175dc6
·
verified ·
1 Parent(s): 7a9fdde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -176,3 +176,10 @@ def get_response(query):
176
  ##############################################
177
  # Classifier
178
  img_size = 224
 
 
 
 
 
 
 
 
176
  ##############################################
177
  # Classifier
178
  img_size = 224
179
+
180
+ @st.cache_resource
181
+ def loadModel():
182
+ model = load_model('models/efficientnet-fine-6.keras')
183
+ return model
184
+
185
+ model = loadModel()