astart01 commited on
Commit
eac90e2
·
verified ·
1 Parent(s): 9d14ef2

Update perv.py

Browse files
Files changed (1) hide show
  1. perv.py +1 -2
perv.py CHANGED
@@ -33,8 +33,7 @@ def run():
33
  preds = self.pipeline.predict(X)
34
  return self.label_encoder.inverse_transform(preds), time.time() - start_time
35
 
36
- with torch.serialization.safe_globals([LabelEncoder]):
37
- checkpoint = torch.load('lstm/model.pt', map_location=torch.device('cpu'), weights_only=False)
38
 
39
  class Attention(nn.Module):
40
  def __init__(self, hidden_dim):
 
33
  preds = self.pipeline.predict(X)
34
  return self.label_encoder.inverse_transform(preds), time.time() - start_time
35
 
36
+ checkpoint = torch.load('models/lstm/model.pt', map_location=torch.device('cpu'))
 
37
 
38
  class Attention(nn.Module):
39
  def __init__(self, hidden_dim):