Update app.py

app.py

@@ -16,6 +16,22 @@ from tensorflow.keras.preprocessing.image import img_to_array
 from huggingface_hub import hf_hub_download
 import gradio as gr
 
+
+MODEL_REPO = "slyviee/img_cap"
+
+# Khởi tạo tài nguyên toàn cục khi app start
+model_path = hf_hub_download(repo_id=MODEL_REPO, filename="best_model.keras")
+tokenizer_path = hf_hub_download(repo_id=MODEL_REPO, filename="tokenizer.pkl")
+config_path = hf_hub_download(repo_id=MODEL_REPO, filename="model_config.pkl")
+
+model = None
+tokenizer = None
+max_length = None
+vocab_size = None
+extractor = None
+ready = False
+startup_error = ""
+
 # -----------------------------
 # Custom attention layers
 # -----------------------------
@@ -48,6 +64,10 @@ class ChannelAttention(layers.Layer):
         config.update({'ratio': self.ratio})
         return config
 
+     @classmethod
+    def from_config(cls, config):
+        return cls(**config)
+
 
 class SpatialAttention(layers.Layer):
     def __init__(self, **kwargs):
@@ -64,6 +84,10 @@ class SpatialAttention(layers.Layer):
     def get_config(self):
         return super(SpatialAttention, self).get_config()
 
+    @classmethod
+    def from_config(cls, config):
+        return cls(**config)
+
 
 # -----------------------------
 # Load model + tokenizer
@@ -75,7 +99,6 @@ def load_caption_model(model_path):
         'SpatialAttention': SpatialAttention
     }
     model = load_model(model_path, custom_objects=custom_objects)
-    print("✅ Đã load model thành công!")
     return model
 
 
@@ -99,7 +122,6 @@ def load_feature_extractor():
 def extract_features_from_image(image_path, extractor):
     image = cv2.imread(image_path)
     if image is None:
-        print(f"❌ Không đọc được ảnh: {image_path}")
         return None
     image = cv2.resize(image, (224, 224))
     image = img_to_array(image)
@@ -131,20 +153,6 @@ def generate_caption(model, tokenizer, image_features, max_length):
 # App initialization
 # -----------------------------
 
-MODEL_REPO = "slyviee/img_cap"
-
-# Khởi tạo tài nguyên toàn cục khi app start
-model_path = hf_hub_download(repo_id=MODEL_REPO, filename="best_model.keras")
-tokenizer_path = hf_hub_download(repo_id=MODEL_REPO, filename="tokenizer.pkl")
-config_path = hf_hub_download(repo_id=MODEL_REPO, filename="model_config.pkl")
-
-model = None
-tokenizer = None
-max_length = None
-vocab_size = None
-extractor = None
-ready = False
-startup_error = ""
 
 def _startup():
     global model, tokenizer, max_length, vocab_size, extractor, ready, startup_error
@@ -188,7 +196,6 @@ def predict(pil_image: Image.Image):
 
 DESCRIPTION = (
     "Upload ảnh và nhận caption sinh ra bởi mô hình. "
-    "Cần có các tệp: best_model.keras, tokenizer.pkl, model_config.pkl."
 )
 
 demo = gr.Interface(