Hugging Face's logo

Spaces:
Setur
/
Marka
Running

App Files Community
unijoh commited on
Commit
c34a3a7
·
verified ·
1 Parent(s): 0da94a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ MODEL_ID = "Setur/BRAGD"
9
  TAGS_FILEPATH = "Sosialurin-GOLD_tags.csv"
10
  HF_TOKEN = os.getenv("BRAGD")
11
 
12
- tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
13
  model = AutoModelForTokenClassification.from_pretrained(MODEL_ID=HF_TOKEN)
14
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
9
  TAGS_FILEPATH = "Sosialurin-GOLD_tags.csv"
10
  HF_TOKEN = os.getenv("BRAGD")
11
 
12
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_ID=HF_TOKEN)
13
  model = AutoModelForTokenClassification.from_pretrained(MODEL_ID=HF_TOKEN)
14
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")