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
|