ksvmuralidhar commited on
Commit
8b93c35
1 Parent(s): bd7da7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -29,7 +29,12 @@ CLASS_INDICES = {'Bean': 0,
29
  'Radish': 13,
30
  'Tomato': 14}
31
 
32
- interpreter = lite.Interpreter(model_path=os.path.join("models", "vegetable_classification_model_mnet.tflite"))
 
 
 
 
 
33
 
34
 
35
  def predict(test_image):
 
29
  'Radish': 13,
30
  'Tomato': 14}
31
 
32
+ @st.cache_resource
33
+ def load_model():
34
+ interpreter = lite.Interpreter(model_path=os.path.join("models", "vegetable_classification_model_mnet.tflite"))
35
+ return interpreter
36
+
37
+ interpreter = load_model()
38
 
39
 
40
  def predict(test_image):