Update demo.py

demo.py

@@ -7,7 +7,7 @@ import os
 os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
 import contextlib
 
-api_key = os.environ.get("api_key")
+
 
 import torch
 
@@ -24,6 +24,8 @@ tqdm.__init__ = partialmethod(tqdm.__init__, leave=False)
 import streamlit as st
 from stmol import *
 
+api_key = st.secrets["api_key"]
+
 def download(outputFile,newFileName,description):
     with open(outputFile, "rb") as file:
         btn = st.download_button(