Spaces:
Running
Running
Simon Dürr
commited on
Commit
•
28cb117
1
Parent(s):
f095f6c
add sample output
Browse files- inference_app.py +40 -3
- test_out.pdb +0 -0
inference_app.py
CHANGED
@@ -6,13 +6,15 @@ import gradio as gr
|
|
6 |
from gradio_molecule3d import Molecule3D
|
7 |
|
8 |
|
|
|
|
|
9 |
def predict (input_sequence, input_ligand):
|
10 |
start_time = time.time()
|
11 |
# Do inference here
|
12 |
# return an output directory
|
13 |
end_time = time.time()
|
14 |
run_time = end_time - start_time
|
15 |
-
return
|
16 |
|
17 |
with gr.Blocks() as app:
|
18 |
|
@@ -25,13 +27,48 @@ with gr.Blocks() as app:
|
|
25 |
|
26 |
# define any options here
|
27 |
|
28 |
-
#
|
29 |
# slider_option = gr.Slider(0,10, label="Slider Option")
|
30 |
# checkbox_option = gr.Checkbox(label="Checkbox Option")
|
31 |
# dropdown_option = gr.Dropdown(["Option 1", "Option 2", "Option 3"], label="Radio Option")
|
32 |
|
33 |
btn = gr.Button("Run Inference")
|
34 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
run_time = gr.Textbox(label="Runtime")
|
36 |
|
37 |
btn.click(predict, inputs=[input_sequence, input_ligand], outputs=[out, run_time])
|
|
|
6 |
from gradio_molecule3d import Molecule3D
|
7 |
|
8 |
|
9 |
+
|
10 |
+
|
11 |
def predict (input_sequence, input_ligand):
|
12 |
start_time = time.time()
|
13 |
# Do inference here
|
14 |
# return an output directory
|
15 |
end_time = time.time()
|
16 |
run_time = end_time - start_time
|
17 |
+
return "test_out.pdb", run_time
|
18 |
|
19 |
with gr.Blocks() as app:
|
20 |
|
|
|
27 |
|
28 |
# define any options here
|
29 |
|
30 |
+
# for automated inference the default options are used
|
31 |
# slider_option = gr.Slider(0,10, label="Slider Option")
|
32 |
# checkbox_option = gr.Checkbox(label="Checkbox Option")
|
33 |
# dropdown_option = gr.Dropdown(["Option 1", "Option 2", "Option 3"], label="Radio Option")
|
34 |
|
35 |
btn = gr.Button("Run Inference")
|
36 |
+
|
37 |
+
gr.Examples(
|
38 |
+
[
|
39 |
+
[
|
40 |
+
"SVKSEYAEAAAVGQEAVAVFNTMKAAFQNGDKEAVAQYLARLASLYTRHEELLNRILEKARREGNKEAVTLMNEFTATFQTGKSIFNAMVAAFKNGDDDSFESYLQALEKVTAKGETLADQIAKAL:SVKSEYAEAAAVGQEAVAVFNTMKAAFQNGDKEAVAQYLARLASLYTRHEELLNRILEKARREGNKEAVTLMNEFTATFQTGKSIFNAMVAAFKNGDDDSFESYLQALEKVTAKGETLADQIAKAL"
|
41 |
+
"COc1ccc(cc1)n2c3c(c(n2)C(=O)N)CCN(C3=O)c4ccc(cc4)N5CCCCC5=O",
|
42 |
+
],
|
43 |
+
],
|
44 |
+
[input_sequence, input_ligand],
|
45 |
+
)
|
46 |
+
reps = [
|
47 |
+
{
|
48 |
+
"model": 0,
|
49 |
+
"chain": "",
|
50 |
+
"resname": "",
|
51 |
+
"style": "cartoon",
|
52 |
+
"color": "whiteCarbon",
|
53 |
+
"residue_range": "",
|
54 |
+
"around": 0,
|
55 |
+
"byres": False,
|
56 |
+
"visible": False
|
57 |
+
},
|
58 |
+
{
|
59 |
+
"model": 1,
|
60 |
+
"chain": "",
|
61 |
+
"resname": "",
|
62 |
+
"style": "stick",
|
63 |
+
"color": "greenCarbon",
|
64 |
+
"residue_range": "",
|
65 |
+
"around": 0,
|
66 |
+
"byres": False,
|
67 |
+
"visible": False
|
68 |
+
}
|
69 |
+
]
|
70 |
+
|
71 |
+
out = Molecule3D(reps=reps)
|
72 |
run_time = gr.Textbox(label="Runtime")
|
73 |
|
74 |
btn.click(predict, inputs=[input_sequence, input_ligand], outputs=[out, run_time])
|
test_out.pdb
ADDED
The diff for this file is too large to render.
See raw diff
|
|