Spaces:
Running
Running
update
Browse files- .gitignore +2 -0
- app.py +3 -7
- 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
|