try hard with api

.gitignore

@@ -0,0 +1,2 @@
+venv/
+.env

app.py

@@ -4,7 +4,7 @@ import io
 import numpy as np
 import os
 import traceback
-import spaces
+# import spaces
 from esm.sdk import client
 from esm.sdk.api import ESM3InferenceClient, ESMProtein, GenerationConfig
 from esm.utils.structure.protein_chain import ProteinChain
@@ -13,6 +13,9 @@ import biotite.structure as bs
 from biotite.structure.io import pdb
 from esm.utils import residue_constants as RC
 
+from dotenv import load_dotenv
+
+load_dotenv()
 # Initialize the model
 token = os.environ.get("ESM_API_TOKEN")
 if not token:
@@ -165,7 +168,7 @@ def visualize_after_pred(protein: ESMProtein, aligned: ESMProtein):
     view.zoomTo()
     return view
 
-@spaces.GPU()
+# @spaces.GPU()
 def run_prediction(pdb_file, num_runs, noise_level, num_frames):
     try:
         if pdb_file is None:

requirements.txt

@@ -7,4 +7,5 @@ pandas
 torch 
 numpy 
 esm
-biotite
+biotite
+python-dotenv