philipobiorah commited on
Commit
bd7368b
·
verified ·
1 Parent(s): cc2fd00

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -1
main.py CHANGED
@@ -22,10 +22,15 @@ app = Flask(__name__)
22
  # Load Model from Hugging Face with Explicit Token
23
  MODEL_HF_REPO = "philipobiorah/bert-imdb-model" # Replace with your Hugging Face model repo
24
 
 
 
 
 
25
  print("🚀 Loading model from Hugging Face Hub with authentication...")
 
26
  model = BertForSequenceClassification.from_pretrained(
27
  MODEL_HF_REPO,
28
- use_auth_token=HF_TOKEN # Pass token explicitly
29
  )
30
 
31
  model.eval()
 
22
  # Load Model from Hugging Face with Explicit Token
23
  MODEL_HF_REPO = "philipobiorah/bert-imdb-model" # Replace with your Hugging Face model repo
24
 
25
+
26
+
27
+
28
+
29
  print("🚀 Loading model from Hugging Face Hub with authentication...")
30
+
31
  model = BertForSequenceClassification.from_pretrained(
32
  MODEL_HF_REPO,
33
+ token=os.getenv("HUGGINGFACE_TOKEN", None)
34
  )
35
 
36
  model.eval()