Update app.py

app.py

@@ -22,7 +22,6 @@ theme = gr.themes.Monochrome(
 )
 
 ## Function to get model from Hugging Face using token
-@spaces.GPU()
 def get_model(model_name, token):
     login(token=token)
 
@@ -185,6 +184,7 @@ 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)