Sid26Roy commited on
Commit
363fc30
·
verified ·
1 Parent(s): 54a5305

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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: "question"}
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)