mawairon commited on
Commit
c9db98a
·
verified ·
1 Parent(s): 3f3c29c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -82,7 +82,7 @@ input_size = 768 + metadata_features # featurizer output size + metadata size
82
  log_reg = LogisticRegressionTorch(input_dim=input_size, output_dim=N_UNIQUE_CLASSES)
83
 
84
  # Load Weights
85
- model_weights_path = '/your_model_weights.pth'
86
  weights = torch.load(model_weights_path, map_location=torch.device('cpu'))
87
 
88
  model = BertClassifier(base_model, log_reg, num_labels = N_UNIQUE_CLASSES)
 
82
  log_reg = LogisticRegressionTorch(input_dim=input_size, output_dim=N_UNIQUE_CLASSES)
83
 
84
  # Load Weights
85
+ model_weights_path = '/gena-blastln-bs33-lr4e-05-S168.pth'
86
  weights = torch.load(model_weights_path, map_location=torch.device('cpu'))
87
 
88
  model = BertClassifier(base_model, log_reg, num_labels = N_UNIQUE_CLASSES)