File size: 1,149 Bytes
d93a43e 0d8eac0 1b279a5 0d8eac0 33e7ec6 0d8eac0 1b279a5 a6fee7b 1b279a5 0029d81 0d8eac0 1ad2f1c 1b279a5 4c70bd8 1b279a5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import gradio as gr
from rdkit.Chem import Draw
from rdkit import Chem
import selfies as sf
def greet1(name):
return name
def greet2(name):
sf_output = ['[C][=C][C][=C][C][=C][Ring1][=Branch1]',
'[C][=C][C][=C][C][=C][C][=C][Ring1][=Branch1]',
'[C][=C][C][=C][C][=C][Ring1][=Branch1][C][=C][C][=C]',
'[C][=C][C][=C][C][=C][Ring1][=Branch1][C@H1][C][=C][C]',
'[C][=C][C][=C][C][=C][Ring1][=Branch1][C@H1][=C][C][=C]']
smis = [sf.decoder(i) for i in sf_output]
mols = []
for smi in smis:
mol = Chem.MolFromSmiles(smi)
mols.append(mol)
img = Draw.MolsToGridImage(
mols,
molsPerRow=4,
subImgSize=(200,200),
legends=['' for x in mols]
)
return img
def greet3(name):
return name+"11"
greeter_1 = gr.Interface(greet1, inputs="text", outputs="text")
greeter_2 = gr.Interface(greet2 , inputs="text", outputs="image")
demo = gr.Parallel(greeter_1, greeter_2)
demo.launch()
#iface = gr.Interface(fn=greet, inputs="text", outputs="image", title="Molecular Language Model as Multi-task Generator",
# )
#iface.launch() |