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 |
+
|