Rename app.py to apps.py

app.py → apps.py

@@ -1,7 +1,6 @@
 from huggingface_hub import login
 from esm.models.esm3 import ESM3
 from esm.sdk.api import ESM3InferenceClient, ESMProtein, GenerationConfig
-import spaces
 import os 
 import gradio as gr
 from gradio_molecule3d import Molecule3D
@@ -14,7 +13,7 @@ TOKEN = os.getenv("HF_TOKEN")
 login(TOKEN)
 
 # This will download the model weights and instantiate the model on your machine.
-model: ESM3InferenceClient = ESM3.from_pretrained("esm3_sm_open_v1").to("cuda")
+model: ESM3InferenceClient = ESM3.from_pretrained("esm3_sm_open_v1").to("cpu")
 
 def read_mol(molpath):
     with open(molpath, "r") as fp:
@@ -76,7 +75,6 @@ def molecule(input_pdb):
     allow-top-navigation-by-user-activation allow-downloads" allowfullscreen="" 
     allowpaymentrequest="" frameborder="0" srcdoc='{x}'></iframe>"""
 
-@spaces.GPU(duration=300)
 def prediction(prompt, temperature):
     protein = ESMProtein(sequence=prompt)
     protein = model.generate(protein, GenerationConfig(track="sequence", num_steps=8, temperature=temperature))