Hugging Face's logo

Spaces:
mawairon
/
NOOTestspace
Sleeping

App Files Community
mawairon commited on
Commit
9ab99fb
·
verified ·
1 Parent(s): 996a1ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -73,7 +73,7 @@ input_size = 768 + metadata_features # featurizer output size + metadata size
73
  log_reg = LogisticRegressionTorch(input_dim=input_size, output_dim=N_UNIQUE_CLASSES)
74
 
75
  # Load Weights
76
- model_weights_path = 'model/gena-blastln-bs33-lr4e-05-S168.pth'
77
  weights = torch.load(model_weights_path, map_location=torch.device('cpu'))
78
 
79
  base_model.load_state_dict(weights['model_state_dict'])
 
73
  log_reg = LogisticRegressionTorch(input_dim=input_size, output_dim=N_UNIQUE_CLASSES)
74
 
75
  # Load Weights
76
+ model_weights_path = 'gena-blastln-bs33-lr4e-05-S168.pth'
77
  weights = torch.load(model_weights_path, map_location=torch.device('cpu'))
78
 
79
  base_model.load_state_dict(weights['model_state_dict'])