update

.gitignore

@@ -0,0 +1,2 @@
+.venv/
+*.pyc

app.py

@@ -98,18 +98,14 @@ with st.sidebar:
 
         st.write(model_name)
     
-# with st.spinner('Setting up the trainer class...'):
-#     trainer = Trainer(ProtConfig())
 
-# with st.spinner('Generating Molecules...'):
-#     trainer.inference()
-
-st.title("text")
 
 if submitted:
     with st.spinner('Creating the trainer class instance...'):
         trainer = Trainer(model_configs[model_name])
-    with st.spinner('Running inference function...'):
+    with st.spinner('Running inference function (this may take a while) ...'):
         trainer.inference()
     st.success(f"Success with the inference of {model_name}")
+else:
+    st.warning("Please select a model to make inference")
 

requirements.txt

@@ -5,4 +5,5 @@ numpy
 seaborn
 matplotlib
 pandas
-torch_geometric
+torch_geometric
+streamlit