Spaces:

battelle
/

FupBERT_Space

Sleeping

c-dunlap commited on Jun 22, 2023

Commit

10d3918

1 Parent(s): 8b58849

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,6 @@ st.set_page_config(layout="wide")
 @st.cache_resource  # Add the caching decorator to prevent model reloads
 def load_model(show_spinner='Loading Model...'):
-    login(token=st.secrets['token'])
     bert_model = AutoModel.from_pretrained('battelle/FupBERT', trust_remote_code=True)
     bert_model.eval()
     return bert_model

 @st.cache_resource  # Add the caching decorator to prevent model reloads
 def load_model(show_spinner='Loading Model...'):
     bert_model = AutoModel.from_pretrained('battelle/FupBERT', trust_remote_code=True)
     bert_model.eval()
     return bert_model