neojex commited on
Commit
9704190
Β·
1 Parent(s): 11adc23

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -27,8 +27,8 @@ class SentimentClassifier(nn.Module):
27
  return self.out(output)
28
  # load the CNN binary classification model
29
  model = SentimentClassifier(len(class_names))
30
- model.load_state_dict(torch.load('model/pytorch_model.bin'))
31
- tokenizer = BertTokenizer.from_pretrained('./model/')
32
 
33
  def encode(text):
34
  encoded_text = tokenizer.encode_plus(
 
27
  return self.out(output)
28
  # load the CNN binary classification model
29
  model = SentimentClassifier(len(class_names))
30
+ model.load_state_dict(torch.load('./pytorch_model.bin'))
31
+ tokenizer = BertTokenizer.from_pretrained('./')
32
 
33
  def encode(text):
34
  encoded_text = tokenizer.encode_plus(