Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import torch.nn.functional as F
|
|
| 6 |
|
| 7 |
@st.cache_resource
|
| 8 |
def load_model():
|
| 9 |
-
absa_tokenizer = AutoTokenizer.from_pretrained("yangheng/deberta-v3-base-absa-v1.1")
|
| 10 |
absa_model = AutoModelForSequenceClassification.from_pretrained("yangheng/deberta-v3-base-absa-v1.1")
|
| 11 |
token_classifier = pipeline(
|
| 12 |
model="thainq107/abte-restaurants-distilbert-base-uncased",
|
|
|
|
| 6 |
|
| 7 |
@st.cache_resource
|
| 8 |
def load_model():
|
| 9 |
+
absa_tokenizer = AutoTokenizer.from_pretrained("yangheng/deberta-v3-base-absa-v1.1", use_fast=False)
|
| 10 |
absa_model = AutoModelForSequenceClassification.from_pretrained("yangheng/deberta-v3-base-absa-v1.1")
|
| 11 |
token_classifier = pipeline(
|
| 12 |
model="thainq107/abte-restaurants-distilbert-base-uncased",
|