Update main.py
Browse files
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 |
-
|
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()
|