Alain Vaucher
commited on
Commit
•
8bbfef5
1
Parent(s):
275093a
Add the pretrained model
Browse files- app.py +3 -1
- pretrained.pt +3 -0
app.py
CHANGED
@@ -18,10 +18,12 @@ logger.addHandler(logging.NullHandler())
|
|
18 |
|
19 |
_SAC_MODEL_TAG = "Heterogenous SAC model (ACE)"
|
20 |
_ORGANIC_MODEL_TAG = "Organic chemistry model"
|
|
|
21 |
|
22 |
model_type_to_models = {
|
23 |
_SAC_MODEL_TAG: ["sac.pt"],
|
24 |
_ORGANIC_MODEL_TAG: ["organic-1.pt", "organic-2.pt", "organic-3.pt"],
|
|
|
25 |
}
|
26 |
|
27 |
SYNTHESIS_TEXT_PLACEHOLDER = (
|
@@ -129,7 +131,7 @@ def launch() -> gr.Interface:
|
|
129 |
title="Extraction of synthesis protocols from paragraphs in machine readable format",
|
130 |
inputs=[
|
131 |
gr.Dropdown(
|
132 |
-
[_SAC_MODEL_TAG, _ORGANIC_MODEL_TAG],
|
133 |
label="Model",
|
134 |
value=_SAC_MODEL_TAG,
|
135 |
),
|
|
|
18 |
|
19 |
_SAC_MODEL_TAG = "Heterogenous SAC model (ACE)"
|
20 |
_ORGANIC_MODEL_TAG = "Organic chemistry model"
|
21 |
+
_PRETRAINED_MODEL_TAG = "Pretrained model"
|
22 |
|
23 |
model_type_to_models = {
|
24 |
_SAC_MODEL_TAG: ["sac.pt"],
|
25 |
_ORGANIC_MODEL_TAG: ["organic-1.pt", "organic-2.pt", "organic-3.pt"],
|
26 |
+
_PRETRAINED_MODEL_TAG: ["pretrained.pt"],
|
27 |
}
|
28 |
|
29 |
SYNTHESIS_TEXT_PLACEHOLDER = (
|
|
|
131 |
title="Extraction of synthesis protocols from paragraphs in machine readable format",
|
132 |
inputs=[
|
133 |
gr.Dropdown(
|
134 |
+
[_SAC_MODEL_TAG, _ORGANIC_MODEL_TAG, _PRETRAINED_MODEL_TAG],
|
135 |
label="Model",
|
136 |
value=_SAC_MODEL_TAG,
|
137 |
),
|
pretrained.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a9fbb1288ea7ac84ec4e46eb2296d45175012a9b968f737c1acbe27cbbb0cc4c
|
3 |
+
size 89958148
|