specollective-ai commited on
Commit
10a9d1c
1 Parent(s): 9efaba0

Update config.

Browse files
Files changed (1) hide show
  1. config.json +18 -20
config.json CHANGED
@@ -1,21 +1,19 @@
1
  {
2
- "sklearn": {
3
- "environment": [
4
- "scikit-learn=1.4.1.post1",
5
- "pandas"
6
- ],
7
- "example_input": {
8
- "data": [
9
- "What is the capital of France?",
10
- "I am feeling very tired today.",
11
- "Please turn off the lights before leaving."
12
- ]
13
- },
14
- "model": {
15
- "file": "custom_model.pkl"
16
- },
17
- "model_format": "pickle",
18
- "task": "text-classification",
19
- "use_intelex": false
20
- }
21
- }
 
1
  {
2
+ "sklearn": {
3
+ "environment": ["scikit-learn=1.4.1.post1", "pandas"],
4
+ "model_type": "regression",
5
+ "example_input": {
6
+ "data": [
7
+ "What is the capital of France?",
8
+ "I am feeling very tired today.",
9
+ "Please turn off the lights before leaving."
10
+ ]
11
+ },
12
+ "model": {
13
+ "file": "custom_model.pkl"
14
+ },
15
+ "model_format": "pickle",
16
+ "task": "text-classification",
17
+ "use_intelex": false
18
+ }
19
+ }