import streamlit as st
from trainer import Trainer

st.title("Hello World")

class DrugGENConfig:
    submodel='CrossLoss'
    act='relu'
    

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

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