Hugging Face's logo

Spaces:
mawairon
/
NOOTestspace
Sleeping

App Files Community
mawairon commited on
Commit
9cef3cf
·
verified ·
1 Parent(s): 285b04a

Upload app.py

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