as-cle-bert commited on
Commit
6e9c18e
1 Parent(s): 0cf7233

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ TOKEN = os.getenv("HF_TOKEN")
13
  login(TOKEN)
14
 
15
  # This will download the model weights and instantiate the model on your machine.
16
- model: ESM3InferenceClient = ESM3.from_pretrained("esm3_sm_open_v1").to("cuda")
17
 
18
  def read_mol(molpath):
19
  with open(molpath, "r") as fp:
 
13
  login(TOKEN)
14
 
15
  # This will download the model weights and instantiate the model on your machine.
16
+ model: ESM3InferenceClient = ESM3.from_pretrained("esm3_sm_open_v1").to("cpu")
17
 
18
  def read_mol(molpath):
19
  with open(molpath, "r") as fp: