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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -3,9 +3,8 @@ from transformers import BertTokenizer, BertForSequenceClassification
3
  import gradio as gr
4
 
5
  # Load model from local folder
6
- model = BertForSequenceClassification.from_pretrained(".", trust_remote_code=True)
7
- tokenizer = BertTokenizer.from_pretrained(".")
8
-
9
  model.eval()
10
  label_map = {0: "statement", 1: "question"}
11
 
 
3
  import gradio as gr
4
 
5
  # Load model from local folder
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