Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = '/
|
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)
|