Spaces:
Running
Running
ThorbenFroehlking
commited on
Commit
·
3e01b86
1
Parent(s):
4eaf58b
Updated
Browse files- .ipynb_checkpoints/app-checkpoint.py +4 -1
- app.py +4 -1
.ipynb_checkpoints/app-checkpoint.py
CHANGED
|
@@ -28,7 +28,10 @@ from datasets import Dataset
|
|
| 28 |
from scipy.special import expit
|
| 29 |
|
| 30 |
# Load model and move to device
|
| 31 |
-
checkpoint = 'ThorbenF/
|
|
|
|
|
|
|
|
|
|
| 32 |
max_length = 1500
|
| 33 |
model, tokenizer = load_model(checkpoint, max_length)
|
| 34 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
|
|
| 28 |
from scipy.special import expit
|
| 29 |
|
| 30 |
# Load model and move to device
|
| 31 |
+
#checkpoint = 'ThorbenF/prot_t5_xl_uniref50'
|
| 32 |
+
#checkpoint = 'ThorbenF/prot_t5_xl_uniref50_cryptic'
|
| 33 |
+
#checkpoint = 'ThorbenF/prot_t5_xl_uniref50_database'
|
| 34 |
+
checkpoint = 'ThorbenF/prot_t5_xl_uniref50_full'
|
| 35 |
max_length = 1500
|
| 36 |
model, tokenizer = load_model(checkpoint, max_length)
|
| 37 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
app.py
CHANGED
|
@@ -28,7 +28,10 @@ from datasets import Dataset
|
|
| 28 |
from scipy.special import expit
|
| 29 |
|
| 30 |
# Load model and move to device
|
| 31 |
-
checkpoint = 'ThorbenF/
|
|
|
|
|
|
|
|
|
|
| 32 |
max_length = 1500
|
| 33 |
model, tokenizer = load_model(checkpoint, max_length)
|
| 34 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
|
|
|
| 28 |
from scipy.special import expit
|
| 29 |
|
| 30 |
# Load model and move to device
|
| 31 |
+
#checkpoint = 'ThorbenF/prot_t5_xl_uniref50'
|
| 32 |
+
#checkpoint = 'ThorbenF/prot_t5_xl_uniref50_cryptic'
|
| 33 |
+
#checkpoint = 'ThorbenF/prot_t5_xl_uniref50_database'
|
| 34 |
+
checkpoint = 'ThorbenF/prot_t5_xl_uniref50_full'
|
| 35 |
max_length = 1500
|
| 36 |
model, tokenizer = load_model(checkpoint, max_length)
|
| 37 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|