import streamlit as st
from trainer import Trainer

st.title("Hello World")

class DrugGENConfig:
    submodel='CrossLoss'
    act='relu'
    z_dim=16
    max_atom=45
    
    

with st.spinner('Setting up the trainer class...'):
    trainer = Trainer(DrugGENConfig())

with st.spinner('Generating Molecules...'):
    trainer.inference()