Hugging Face's logo

Spaces:
omarperacha
/
protein-secondary-structure-prediction
Runtime error

App Files Community
omarperacha commited on
Commit
4c9631f
·
1 Parent(s): 5a79fe4

get ss seq

Browse files
Files changed (1) hide show
  1. ps4_eval/eval.py +1 -1
ps4_eval/eval.py CHANGED
@@ -33,7 +33,7 @@ def sample_new_sequence(embs, weights_load_path, model_name='PS4_Mega'):
33
  model = load_trained_model(weights_load_path, model_name)
34
 
35
  seq_size = len(embs)
36
- R = embs.view(1, seq_size)
37
 
38
  pred_ss = ''
39
 
 
33
  model = load_trained_model(weights_load_path, model_name)
34
 
35
  seq_size = len(embs)
36
+ R = embs.view(1, seq_size, -1)
37
 
38
  pred_ss = ''
39