Spaces:
Runtime error
Runtime error
Commit
·
dc3d11f
1
Parent(s):
a388d0d
updated config
Browse files- config.yaml +14 -0
config.yaml
CHANGED
@@ -0,0 +1,14 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
SENTIMENT_CLF:
|
2 |
+
sent_chkpt: 'distilbert-base-uncased-finetuned-sst-2-english'
|
3 |
+
sent_mdl_dir: 'sentiment_model_dir'
|
4 |
+
sent_onnx_mdl_dir: 'sent_clf_onnx_dir'
|
5 |
+
sent_onnx_mdl_name: 'sentiment_classifier_onnx.onnx'
|
6 |
+
sent_onnx_quant_mdl_name: 'sentiment_classifier_onnx_quant.onnx'
|
7 |
+
|
8 |
+
ZEROSHOT_CLF:
|
9 |
+
zs_chkpt: 'valhalla/distilbart-mnli-12-1'
|
10 |
+
zs_mdl_dir: 'zs_model_dir'
|
11 |
+
zs_onnx_mdl_dir: 'zs_onnx_dir'
|
12 |
+
zs_onnx_mdl_name: 'model.onnx'
|
13 |
+
zs_onnx_quant_mdl_name: 'model_quant.onnx'
|
14 |
+
|