Update app.py

app.py

@@ -108,9 +108,14 @@ if submitted:
     with st.spinner(f'Creating the trainer class instance for {model_name}...'):
         trainer = Trainer(config)
     with st.spinner(f'Running inference function of {model_name} (this may take a while) ...'):
-        trainer.inference()
-    st.success(f"Success with the inference of {model_name}")
+        results = trainer.inference()
+    st.success(f"Inference of {model_name} took {results["runtime"]:.2f} seconds.")
 
+    with st.expander("Expand to see scores"):
+
+        st.write(f"Fraction valid: {results["fraction_valid"]}")
+        st.write(f"Uniqueness: {results["uniqueness"]}")
+        st.write(f"Novelty score: {results["novelty"]}")
 
     with open(f'experiments/inference/{model_name}/inference_drugs.txt') as f:
         inference_drugs = f.read()