bcadkins01 commited on
Commit
af1913b
·
verified ·
1 Parent(s): e295393

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ st.set_page_config(page_title='Beta-Lactam Molecule Generator', layout='wide')
18
  @st.cache_resource(show_spinner="Loading Models...", ttl=600)
19
  def load_models():
20
  # Load your molecule generation model
21
- model_name = "your-new-beta-lactam-model-path" # Replace with your actual model path
22
  access_token = os.getenv("HUGGING_FACE_TOKEN")
23
  model = BartForConditionalGeneration.from_pretrained(model_name, use_auth_token=access_token)
24
  tokenizer = BartTokenizer.from_pretrained(model_name, use_auth_token=access_token)
 
18
  @st.cache_resource(show_spinner="Loading Models...", ttl=600)
19
  def load_models():
20
  # Load your molecule generation model
21
+ model_name = "bcadkins01/beta_lactam_generator" # Replace with your actual model path
22
  access_token = os.getenv("HUGGING_FACE_TOKEN")
23
  model = BartForConditionalGeneration.from_pretrained(model_name, use_auth_token=access_token)
24
  tokenizer = BartTokenizer.from_pretrained(model_name, use_auth_token=access_token)