Added HF_TOKEN

app.py

@@ -2,6 +2,7 @@ import gradio
 import logging
 import time
 from transformers import pipeline
+import os
 
 logger = logging.getLogger("gradio_test_001")
 logger.setLevel(logging.INFO)
@@ -13,11 +14,12 @@ categories = [
     "Card Payments", "Banking", "Regulations", "Account Payments"
 ]
 logging.debug("Categories to classify: " + repr(categories))
-
+HF_TOKEN = os.getenv("HF_TOKEN")
 classifier = pipeline("zero-shot-classification",
                       # model="google-bert/bert-base-uncased")
-                      model="Xylor/bert-case-uncased-single-sampled")
                       # model="facebook/bart-large-mnli")
+                      model="Xylor/bert-case-uncased-single-sampled",
+                      token=HF_TOKEN)
 
 # sequence_to_classify = "one day I will see the world"
 # candidate_labels = ['travel', 'cooking', 'dancing']