Hugging Face's logo

Spaces:
ThorbenF
/
test_webpage
Sleeping

App Files Community
ThorbenF commited on
Commit
0efae95
·
1 Parent(s): b081fc7

Update

Browse files
Files changed (2) hide show
  1. .ipynb_checkpoints/app-checkpoint.py +2 -1
  2. app.py +2 -1
.ipynb_checkpoints/app-checkpoint.py CHANGED
@@ -29,7 +29,8 @@ from scipy.special import expit
29
 
30
 
31
  # Load model and move to device
32
- checkpoint = 'ThorbenF/prot_t5_xl_uniref50'
 
33
  max_length = 1500
34
  model, tokenizer = load_model(checkpoint, max_length)
35
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
29
 
30
 
31
  # Load model and move to device
32
+ #checkpoint = 'ThorbenF/prot_t5_xl_uniref50'
33
+ checkpoint = 'ThorbenF/prot_t5_xl_uniref50_cryptic'
34
  max_length = 1500
35
  model, tokenizer = load_model(checkpoint, max_length)
36
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
app.py CHANGED
@@ -29,7 +29,8 @@ from scipy.special import expit
29
 
30
 
31
  # Load model and move to device
32
- checkpoint = 'ThorbenF/prot_t5_xl_uniref50'
 
33
  max_length = 1500
34
  model, tokenizer = load_model(checkpoint, max_length)
35
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
29
 
30
 
31
  # Load model and move to device
32
+ #checkpoint = 'ThorbenF/prot_t5_xl_uniref50'
33
+ checkpoint = 'ThorbenF/prot_t5_xl_uniref50_cryptic'
34
  max_length = 1500
35
  model, tokenizer = load_model(checkpoint, max_length)
36
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')