Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
|
|
6 |
model = BertForSequenceClassification.from_pretrained("bert-expense-classifier/bert-expense-classifier", trust_remote_code=True)
|
7 |
tokenizer = BertTokenizer.from_pretrained("bert-expense-classifier/bert-expense-classifier")
|
8 |
model.eval()
|
9 |
-
label_map = {0: "statement", 1: "
|
10 |
|
11 |
def classify_sentence(text):
|
12 |
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=128)
|
|
|
6 |
model = BertForSequenceClassification.from_pretrained("bert-expense-classifier/bert-expense-classifier", trust_remote_code=True)
|
7 |
tokenizer = BertTokenizer.from_pretrained("bert-expense-classifier/bert-expense-classifier")
|
8 |
model.eval()
|
9 |
+
label_map = {0: "statement", 1: "query"}
|
10 |
|
11 |
def classify_sentence(text):
|
12 |
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=128)
|