Update app.py

app.py

@@ -1,6 +1,19 @@
 import streamlit as st
-
+from trainer import Trainer
 
 st.title("Hello World")
 
-st.write("This is a simple example of Streamlit")
+class DrugGENConfig:
+    submodel='CrossLoss'
+    act='relu'
+    
+
+with st.spinner('Setting up the trainer class...'):
+    trainer = Trainer(config)
+
+with st.spinner('Generating Molecules...'):
+    trainer.inference()
+
+
+
+