Anton Bushuiev commited on
Commit
7d3d4d7
·
1 Parent(s): 4baed9c

Update function to download from zenodo

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -38,7 +38,7 @@ from ppiref.extraction import PPIExtractor
38
  from ppiref.utils.ppi import PPIPath
39
  from ppiref.utils.residue import Residue
40
  from ppiformer.tasks.node import DDGPPIformer
41
- from ppiformer.utils.api import download_weights
42
  from ppiformer.utils.api import predict_ddg as predict_ddg_
43
  from ppiformer.utils.torch import fill_diagonal
44
  from ppiformer.definitions import PPIFORMER_WEIGHTS_DIR
@@ -484,7 +484,7 @@ with app:
484
  gr.Image("assets/logos.png")
485
 
486
  # Download weights from Zenodo
487
- download_weights()
488
 
489
  # Set device
490
  # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
38
  from ppiref.utils.ppi import PPIPath
39
  from ppiref.utils.residue import Residue
40
  from ppiformer.tasks.node import DDGPPIformer
41
+ from ppiformer.utils.api import download_from_zenodo
42
  from ppiformer.utils.api import predict_ddg as predict_ddg_
43
  from ppiformer.utils.torch import fill_diagonal
44
  from ppiformer.definitions import PPIFORMER_WEIGHTS_DIR
 
484
  gr.Image("assets/logos.png")
485
 
486
  # Download weights from Zenodo
487
+ download_from_zenodo('weights.zip')
488
 
489
  # Set device
490
  # device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')