phuongsuga commited on
Commit
76c71df
·
verified ·
1 Parent(s): dc37c6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -3
app.py CHANGED
@@ -24,16 +24,26 @@ THRESHOLD = 0.65
24
  # LOAD TEXT MODEL
25
  # =====================
26
  print("🔹 Downloading text model...")
27
- tokenizer = AutoTokenizer.from_pretrained(f"{TEXT_MODEL_REPO}/text_model", use_fast=False)
28
- text_model = AutoModelForSequenceClassification.from_pretrained(f"{TEXT_MODEL_REPO}/text_model/checkpoint-3390")
 
 
 
 
 
 
 
 
 
 
29
  text_model.to(DEVICE).eval()
30
 
31
  # Load label2id.json từ repo
32
- import requests
33
  label_url = f"https://huggingface.co/{TEXT_MODEL_REPO}/resolve/main/text_model/label2id.json"
34
  label2id = requests.get(label_url).json()
35
  id2label_text = {i: l for l, i in label2id.items()}
36
 
 
37
  # =====================
38
  # LOAD IMAGE MODEL
39
  # =====================
 
24
  # LOAD TEXT MODEL
25
  # =====================
26
  print("🔹 Downloading text model...")
27
+
28
+ tokenizer = AutoTokenizer.from_pretrained(
29
+ TEXT_MODEL_REPO,
30
+ subfolder="text_model",
31
+ use_fast=False
32
+ )
33
+
34
+ text_model = AutoModelForSequenceClassification.from_pretrained(
35
+ TEXT_MODEL_REPO,
36
+ subfolder="text_model/checkpoint-3390"
37
+ )
38
+
39
  text_model.to(DEVICE).eval()
40
 
41
  # Load label2id.json từ repo
 
42
  label_url = f"https://huggingface.co/{TEXT_MODEL_REPO}/resolve/main/text_model/label2id.json"
43
  label2id = requests.get(label_url).json()
44
  id2label_text = {i: l for l, i in label2id.items()}
45
 
46
+
47
  # =====================
48
  # LOAD IMAGE MODEL
49
  # =====================