Update app.py

app.py

@@ -5,7 +5,7 @@ import os, tempfile
 import torch
 import py3Dmol
 from huggingface_hub import login
-import spaces
+# import spaces
 
 from esm.utils.structure.protein_chain import ProteinChain
 from esm.models.esm3 import ESM3
@@ -22,6 +22,7 @@ theme = gr.themes.Monochrome(
 )
 
 ## Function to get model from Hugging Face using token
+# @spaces.GPU()
 def get_model(model_name, token):
     login(token=token)
 
@@ -34,7 +35,6 @@ 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())]