Spaces:

hai2131
/

ABSA_GCN

Sleeping

hai2131 commited on 18 days ago

Commit

c3a28c2

verified ·

1 Parent(s): 27fbc65

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch.nn.functional as F
 @st.cache_resource
 def load_model():
-    absa_tokenizer = AutoTokenizer.from_pretrained("yangheng/deberta-v3-base-absa-v1.1")
     absa_model = AutoModelForSequenceClassification.from_pretrained("yangheng/deberta-v3-base-absa-v1.1")
     token_classifier = pipeline(
         model="thainq107/abte-restaurants-distilbert-base-uncased",

 @st.cache_resource
 def load_model():
+    absa_tokenizer = AutoTokenizer.from_pretrained("yangheng/deberta-v3-base-absa-v1.1", use_fast=False)
     absa_model = AutoModelForSequenceClassification.from_pretrained("yangheng/deberta-v3-base-absa-v1.1")
     token_classifier = pipeline(
         model="thainq107/abte-restaurants-distilbert-base-uncased",