Sirapatsorn commited on
Commit
58f8925
·
verified ·
1 Parent(s): 9764c40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ import torch
3
  from transformers import BertTokenizer, BertForSequenceClassification, XLNetTokenizer, XLNetForSequenceClassification
4
 
5
  # โหลดโมเดลของคุณ
6
- logbert_model = BertForSequenceClassification.from_pretrained("Sirapatsorn/Spark_Log_Analysis", use_auth_token=True)
7
- xlnet_model = XLNetForSequenceClassification.from_pretrained("Sirapatsorn/Spark_Log_Analysis", use_auth_token=True)
8
  logbert_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
9
  xlnet_tokenizer = XLNetTokenizer.from_pretrained("xlnet-base-cased")
10
 
 
3
  from transformers import BertTokenizer, BertForSequenceClassification, XLNetTokenizer, XLNetForSequenceClassification
4
 
5
  # โหลดโมเดลของคุณ
6
+ logbert_model = BertForSequenceClassification.from_pretrained("Sirapatsorn/Spark_Log_Analysis", token=os.getenv("HUGGINGFACE_TOKEN"))
7
+ xlnet_model = XLNetForSequenceClassification.from_pretrained("Sirapatsorn/Spark_Log_Analysis", token=os.getenv("HUGGINGFACE_TOKEN"))
8
  logbert_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
9
  xlnet_tokenizer = XLNetTokenizer.from_pretrained("xlnet-base-cased")
10