Spaces:

sohomghosh
/

FinLanSer_Financial_Language_Simplifier

Runtime error

sohomju commited on Jan 24, 2024

Commit

fac3fce

verified ·

1 Parent(s): b9eb22d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ tokenizer_read = BertTokenizer.from_pretrained('ProsusAI/finbert')
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_read = BERTClass(2, "readability")
 model_read.to(device)
-model_read.load_state_dict(torch.load('readability_model.bin', map_location=device)['model_state_dict'])
 def get_readability(text):

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_read = BERTClass(2, "readability")
 model_read.to(device)
+model_read.load_state_dict(torch.load('readability_model.bin', map_location=device, strict=False)['model_state_dict'])
 def get_readability(text):