Update app.py

app.py

@@ -76,14 +76,27 @@ class NoTargetConfig(DrugGENConfig):
     submodel="NoTarget"
     inference_model="experiments/models/NoTarget"
 
+
+model_configs = {
+    "Prot": ProtConfig(),
+    "CrossLoss": CrossLossConfig(),
+    "NoTarget": NoTargetConfig(),
+}
+
+
 with st.sidebar:
     st.title("DrugGEN")
+    with st.form("model_selection_from"):
 
-    model_name = st.radio(
-    "Select a model to make inference",
-    ('Prot', 'CrossLoss', 'NoTarget'))
+        model_name = st.radio(
+        "Select a model to make inference",
+        ('Prot', 'CrossLoss', 'NoTarget'))
 
-    st.write(model_name)
+        submitted = st.form_submit_button("Start Computing")
+
+    if submitted:
+
+        st.write(model_name)
     
 # with st.spinner('Setting up the trainer class...'):
 #     trainer = Trainer(ProtConfig())
@@ -93,5 +106,8 @@ with st.sidebar:
 
 st.title("text")
 
-
+if submitted:
+    trainer = Trainer(model_configs[model_name])
+    trainer.inference()
+    st.success(f"Success with the inference of {model_name}")