Hugging Face's logo

Spaces:
wangjin2000
/
ESM2PPI
Paused

App Files Community
wangjin2000 commited on
Commit
4d87697
·
verified ·
1 Parent(s): bd84d59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -29,8 +29,10 @@ PEFT_MODEL_OPTIONS = [
29
  "wangjin2000/esm2_t6_8M_UR50D_PPI_2024-09-19_06",
30
  "wangjin2000/esm2_t12_35M_UR50D_PPI_2024-09-19_09",
31
  "wangjin2000/esm2_t33_650M_UR50D_PPI_2024-09-19_10",
32
- "TianlaiChen/PepMLM-650M",
33
- ] # finetuned models
 
 
34
 
35
  #build datasets
36
  class ProteinDataset(Dataset):
 
29
  "wangjin2000/esm2_t6_8M_UR50D_PPI_2024-09-19_06",
30
  "wangjin2000/esm2_t12_35M_UR50D_PPI_2024-09-19_09",
31
  "wangjin2000/esm2_t33_650M_UR50D_PPI_2024-09-19_10",
32
+ "facebook/esm2_t6_8M_UR50D",
33
+ "facebook/esm2_t12_35M_UR50D",
34
+ "facebook/esm2_t33_650M_UR50D",
35
+ ] # finetuned models, but the last three models are original fundation models
36
 
37
  #build datasets
38
  class ProteinDataset(Dataset):