Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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",
|
7 |
-
xlnet_model = XLNetForSequenceClassification.from_pretrained("Sirapatsorn/Spark_Log_Analysis",
|
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 |
|