Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -109,13 +109,13 @@ if submitted:
|
|
| 109 |
trainer = Trainer(config)
|
| 110 |
with st.spinner(f'Running inference function of {model_name} (this may take a while) ...'):
|
| 111 |
results = trainer.inference()
|
| 112 |
-
st.success(f"Inference of {model_name} took {results[
|
| 113 |
|
| 114 |
with st.expander("Expand to see scores"):
|
| 115 |
|
| 116 |
-
st.write(f"Fraction valid: {results[
|
| 117 |
-
st.write(f"Uniqueness: {results[
|
| 118 |
-
st.write(f"Novelty score: {results[
|
| 119 |
|
| 120 |
with open(f'experiments/inference/{model_name}/inference_drugs.txt') as f:
|
| 121 |
inference_drugs = f.read()
|
|
|
|
| 109 |
trainer = Trainer(config)
|
| 110 |
with st.spinner(f'Running inference function of {model_name} (this may take a while) ...'):
|
| 111 |
results = trainer.inference()
|
| 112 |
+
st.success(f"Inference of {model_name} took {results['runtime']:.2f} seconds.")
|
| 113 |
|
| 114 |
with st.expander("Expand to see scores"):
|
| 115 |
|
| 116 |
+
st.write(f"Fraction valid: {results['fraction_valid']}")
|
| 117 |
+
st.write(f"Uniqueness: {results['uniqueness']}")
|
| 118 |
+
st.write(f"Novelty score: {results['novelty']}")
|
| 119 |
|
| 120 |
with open(f'experiments/inference/{model_name}/inference_drugs.txt') as f:
|
| 121 |
inference_drugs = f.read()
|