Update app.py

app.py

@@ -6,10 +6,13 @@ 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(config)
+    trainer = Trainer(DrugGENConfig())
 
 with st.spinner('Generating Molecules...'):
     trainer.inference()