Update app.py

app.py

@@ -34,6 +34,7 @@ def get_model(model_name, token):
     return model
 
 ## Function to get PDB data
+@spaces.GPU()
 def get_pdb(pdb_id, chain_id):
     pdb = ProteinChain.from_rcsb(pdb_id, chain_id)
     # return [pdb.sequence, render_pdb(pdb.to_pdb_string())]
@@ -184,7 +185,6 @@ def ss_edit(model_name, token, pdb_id, chain_id, region_start, region_end, short
         ]
 
 ## Function for SASA Editing
-@spaces.GPU()
 def sasa_edit(model_name, token, pdb_id, chain_id, span_start, span_end, n_samples):
     pdb = get_pdb(pdb_id, chain_id)