Hugging Face's logo

Spaces:
LinguaCustodia
/
ClassiFin-demo
Sleeping

App Files Community
gcaillaut commited on
Commit
ae2aca3
·
1 Parent(s): 0d9ce7f

update tokenizer

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -4,9 +4,8 @@ import torch
4
 
5
  DEVICE = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
6
  MODEL_ID = "LinguaCustodia/ClassiFin"
7
- TOKENIZER_ID = "Snowflake/snowflake-arctic-embed-m"
8
  MODEL = AutoModelForSequenceClassification.from_pretrained(MODEL_ID).to(DEVICE)
9
- TOKENIZER = AutoTokenizer.from_pretrained(TOKENIZER_ID)
10
 
11
 
12
  def classify(text):
 
4
 
5
  DEVICE = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
6
  MODEL_ID = "LinguaCustodia/ClassiFin"
 
7
  MODEL = AutoModelForSequenceClassification.from_pretrained(MODEL_ID).to(DEVICE)
8
+ TOKENIZER = AutoTokenizer.from_pretrained(MODEL_ID)
9
 
10
 
11
  def classify(text):