Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "
|
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)
|