Update app.py

app.py

@@ -8,6 +8,7 @@ from tensorflow.keras.utils import load_img, save_img, img_to_array
 from numpy import array, round
 from utils.delete_file import delete_file
 from requests import get
+from functools import lru_cache
 
 
 app = Flask(__name__, template_folder='templates', static_folder='static')
@@ -29,12 +30,13 @@ CLASS_INDICES = {'Bean': 0,
                  'Radish': 13,
                  'Tomato': 14}
 
-@st.cache_resource
+@lru_cache(max_size=1)
 def load_model():
     interpreter = lite.Interpreter(model_path=os.path.join("models", "vegetable_classification_model_mnet.tflite"))
+    print('loaded model')
     return interpreter
 
-interpreter = load_model()
+interpreter = load_model(True)
 
 
 def predict(test_image):