Update app/utils/predict_afteruse.py

app/utils/predict_afteruse.py

@@ -1,10 +1,14 @@
 import torch
 from transformers import BertTokenizer, BertForSequenceClassification
 import numpy as np
+import os
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# Load the pre-trained model and tokenizer
+tokenizer = BertTokenizer.from_pretrained("Maulidaaa/bert-safe-multilabel", token=HF_TOKEN)
+model = BertForSequenceClassification.from_pretrained("Maulidaaa/bert-safe-multilabel", token=HF_TOKEN)
 
-# Load model dan tokenizer multi-label
-tokenizer = BertTokenizer.from_pretrained(r"D:\python\pythonProject\skicare_analyze\bert-safe-multilabel", local_files_only=True)
-model = BertForSequenceClassification.from_pretrained(r"D:\python\pythonProject\skicare_analyze\bert-safe-multilabel", local_files_only=True)
 model.eval()
 
 # Efek after use (harus sesuai saat training)