Update app.py

app.py

@@ -77,7 +77,13 @@ class NoTargetConfig(DrugGENConfig):
     inference_model="experiments/models/NoTarget"
 
 with st.sidebar:
-    st.success("test")
+    st.title("DrugGEN")
+
+    model_name = st.radio(
+    "Select a model to make inference",
+    ('Prot', 'CrossLoss', 'NoTarget'))
+
+    st.write(model_name)
     
 # with st.spinner('Setting up the trainer class...'):
 #     trainer = Trainer(ProtConfig())
@@ -85,6 +91,7 @@ with st.sidebar:
 # with st.spinner('Generating Molecules...'):
 #     trainer.inference()
 
+st.title("text")