sohomghosh commited on
Commit
d08b0ab
1 Parent(s): b9eb22d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ tokenizer_read = BertTokenizer.from_pretrained('ProsusAI/finbert')
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  model_read = BERTClass(2, "readability")
17
  model_read.to(device)
18
- model_read.load_state_dict(torch.load('readability_model.bin', map_location=device)['model_state_dict'])
19
 
20
 
21
  def get_readability(text):
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  model_read = BERTClass(2, "readability")
17
  model_read.to(device)
18
+ model_read.load_state_dict(torch.load('readability_model.bin', map_location=device)['model_state_dict'], strict=False)
19
 
20
 
21
  def get_readability(text):