Update app.py

app.py

@@ -5,6 +5,9 @@ from pathlib import Path
 
 import gradio as gr
 from PIL import Image
+from huggingface_hub import hf_hub_download
+
+MODEL_REPO = "slyviee/img_cap"
 
 # Import các hàm từ source gốc
 from flickr30k import (
@@ -16,9 +19,9 @@ from flickr30k import (
 )
 
 # Khởi tạo tài nguyên toàn cục khi app start
-MODEL_PATH = os.environ.get("CAP_MODEL_PATH", "best_model.keras")
-TOKENIZER_PATH = os.environ.get("CAP_TOKENIZER_PATH", "tokenizer.pkl")
-CONFIG_PATH = os.environ.get("CAP_CONFIG_PATH", "model_config.pkl")
+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
@@ -70,7 +73,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(