Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import torch
 import numpy as np
@@ -6,9 +7,10 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification
 
 MODEL_ID = "Setur/BRAGD"
 TAGS_FILEPATH = "Sosialurin-GOLD_tags.csv"
+HF_TOKEN = os.getenv("BRAGD")
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
-model = AutoModelForTokenClassification.from_pretrained(MODEL_ID)
+model = AutoModelForTokenClassification.from_pretrained(MODEL_ID=HF_TOKEN)
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)