improving layout
Browse files
app.py
CHANGED
@@ -9,6 +9,7 @@ from AtomLenz import *
|
|
9 |
#from utils_graph import *
|
10 |
from Object_Smiles import Objects_Smiles
|
11 |
from rdkit import Chem
|
|
|
12 |
#from robust_detection import wandb_config
|
13 |
from robust_detection import utils
|
14 |
from robust_detection.models.rcnn import RCNN
|
@@ -203,7 +204,7 @@ def atomlenz(modelfile):
|
|
203 |
if m is None:
|
204 |
st.write("No valid chemical structure recognized")
|
205 |
else:
|
206 |
-
img =
|
207 |
st.image(img, caption=f"{smiles}", use_column_width=True)
|
208 |
print(pred)
|
209 |
|
|
|
9 |
#from utils_graph import *
|
10 |
from Object_Smiles import Objects_Smiles
|
11 |
from rdkit import Chem
|
12 |
+
from rdkit.Chem import Draw
|
13 |
#from robust_detection import wandb_config
|
14 |
from robust_detection import utils
|
15 |
from robust_detection.models.rcnn import RCNN
|
|
|
204 |
if m is None:
|
205 |
st.write("No valid chemical structure recognized")
|
206 |
else:
|
207 |
+
img = Draw.MolToImage(m)
|
208 |
st.image(img, caption=f"{smiles}", use_column_width=True)
|
209 |
print(pred)
|
210 |
|