Hugging Face's logo

Spaces:
sleiyer
/
restricted_item_detector
Sleeping

App Files Community
Sheng Lei commited on
Commit
9d5283f
•
1 Parent(s): 99b5f7f

update syntax

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from transformers import BertTokenizer, BertForSequenceClassification
11
  import torch
12
 
13
 
14
- model = BertForSequenceClassification.load("sleiyer/restricted_item_detector")
15
  # Load the trained model and tokenizer
16
  tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
17
 
 
11
  import torch
12
 
13
 
14
+ model = BertForSequenceClassification.from_pretrained("sleiyer/restricted_item_detector")
15
  # Load the trained model and tokenizer
16
  tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
17