add cache data

app.py

@@ -9,8 +9,13 @@ from tensorflow.keras.models import load_model
 from tensorflow.keras.preprocessing import image
 
 UTC_8 = pytz.timezone('Asia/Shanghai')
-#Load the model 
-my_model = load_model("model/zha2024_6.h5")
+#cahe func
+@st.cache_resource
+def load_my_model():
+    model = load_model("model/zha2024_6.h5")
+    return model
+    
+my_model = load_my_model()
 target_size = (300, 300)
 class_labels = {0: '炭黑组', 1: '正常发挥', 2: '炫彩组', 3: '糊糊组', 4: '炸组日常', 5: '凡尔赛',6: '非食物'}
 predicted_class=''
@@ -52,7 +57,8 @@ safety_settings = [
 ]
 
 #fuctions
-def preprocess_image(img_path):
+@st.cache_data
+def preprocess_image_cached(img_path, target_size):
     img = image.load_img(img_path, target_size=target_size)
     img_array = image.img_to_array(img)
     img_array = np.expand_dims(img_array, axis=0)
@@ -83,7 +89,7 @@ def get_critic_info(review_style):
 
 def img_score(img_raw_path):
     global predicted_class
-    img_array = preprocess_image(img_raw_path)
+    img_array = preprocess_image_cached(img_raw_path, target_size)
     predictions = my_model.predict(img_array)
     predicted_class_index = np.argmax(predictions, axis=-1)
     predicted_class = class_labels[predicted_class_index[0]]