osbm commited on
Commit
26b55ee
·
1 Parent(s): e7c4633
Files changed (3) hide show
  1. .gitignore +2 -0
  2. app.py +3 -7
  3. requirements.txt +2 -1
.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ .venv/
2
+ *.pyc
app.py CHANGED
@@ -98,18 +98,14 @@ with st.sidebar:
98
 
99
  st.write(model_name)
100
 
101
- # with st.spinner('Setting up the trainer class...'):
102
- # trainer = Trainer(ProtConfig())
103
 
104
- # with st.spinner('Generating Molecules...'):
105
- # trainer.inference()
106
-
107
- st.title("text")
108
 
109
  if submitted:
110
  with st.spinner('Creating the trainer class instance...'):
111
  trainer = Trainer(model_configs[model_name])
112
- with st.spinner('Running inference function...'):
113
  trainer.inference()
114
  st.success(f"Success with the inference of {model_name}")
 
 
115
 
 
98
 
99
  st.write(model_name)
100
 
 
 
101
 
 
 
 
 
102
 
103
  if submitted:
104
  with st.spinner('Creating the trainer class instance...'):
105
  trainer = Trainer(model_configs[model_name])
106
+ with st.spinner('Running inference function (this may take a while) ...'):
107
  trainer.inference()
108
  st.success(f"Success with the inference of {model_name}")
109
+ else:
110
+ st.warning("Please select a model to make inference")
111
 
requirements.txt CHANGED
@@ -5,4 +5,5 @@ numpy
5
  seaborn
6
  matplotlib
7
  pandas
8
- torch_geometric
 
 
5
  seaborn
6
  matplotlib
7
  pandas
8
+ torch_geometric
9
+ streamlit