Alain Vaucher commited on
Commit
8bbfef5
1 Parent(s): 275093a

Add the pretrained model

Browse files
Files changed (2) hide show
  1. app.py +3 -1
  2. 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