Update app.py

app.py

@@ -15,8 +15,8 @@ from flask_cors import CORS
 # =====================
 # CONFIG
 # =====================
-TEXT_MODEL_REPO = "phuongsuga/PBL6_AI_Model_Text_Image"  # repo chứa mô hình text
-IMAGE_MODEL_REPO = "phuongsuga/PBL6_AI_Model_Image"      # repo chứa mô hình image
+TEXT_MODEL_REPO = "phuongsuga/PBL6_AI_Model_Text_Image"
+IMAGE_MODEL_REPO = "phuongsuga/PBL6_AI_Model_Image"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 THRESHOLD = 0.65
 
@@ -24,12 +24,13 @@ THRESHOLD = 0.65
 # LOAD TEXT MODEL
 # =====================
 print("🔹 Downloading text model...")
-tokenizer = AutoTokenizer.from_pretrained(TEXT_MODEL_REPO, use_fast=False)
-text_model = AutoModelForSequenceClassification.from_pretrained(TEXT_MODEL_REPO)
+tokenizer = AutoTokenizer.from_pretrained(f"{TEXT_MODEL_REPO}/text_model", use_fast=False)
+text_model = AutoModelForSequenceClassification.from_pretrained(f"{TEXT_MODEL_REPO}/text_model/checkpoint-3390")
 text_model.to(DEVICE).eval()
 
-# Load label2id.json
-label_url = f"https://huggingface.co/{TEXT_MODEL_REPO}/resolve/main/label2id.json"
+# Load label2id.json từ repo
+import requests
+label_url = f"https://huggingface.co/{TEXT_MODEL_REPO}/resolve/main/text_model/label2id.json"
 label2id = requests.get(label_url).json()
 id2label_text = {i: l for l, i in label2id.items()}
 
@@ -43,13 +44,13 @@ def build_model(num_classes=4):
     return timm.create_model("efficientnet_b3", pretrained=False, num_classes=num_classes)
 
 image_model = build_model()
+image_model_path = f"https://huggingface.co/{IMAGE_MODEL_REPO}/resolve/main/image_model/efficientnet_b3.pth"
 
-# tải file weight từ repo Hugging Face
-image_model_path = f"https://huggingface.co/{IMAGE_MODEL_REPO}/resolve/main/efficientnet_b3.pth"
 torch.hub.download_url_to_file(image_model_path, "efficientnet_b3.pth")
 image_model.load_state_dict(torch.load("efficientnet_b3.pth", map_location=DEVICE))
 image_model.to(DEVICE).eval()
 
+
 # Chuẩn hoá ảnh đầu vào
 val_transforms = transforms.Compose([
     transforms.Resize((224, 224)),