data-silence
commited on
Commit
•
264a654
1
Parent(s):
5d75a68
add config.json
Browse files- config.json +0 -5
- inference.py +2 -1
config.json
DELETED
@@ -1,5 +0,0 @@
|
|
1 |
-
{
|
2 |
-
"model_type": "fasttext",
|
3 |
-
"fasttext_version": "0.9.2",
|
4 |
-
"labels": ["climate", "conflicts", "culture", "economy", "gloss", "health", "politics", "science", "society", "sports", "travel"]
|
5 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
inference.py
CHANGED
@@ -20,4 +20,5 @@ class FastTextClassifierPipeline(pipeline):
|
|
20 |
return results
|
21 |
|
22 |
|
23 |
-
|
|
|
|
20 |
return results
|
21 |
|
22 |
|
23 |
+
def pipeline(task="text-classification", model=None):
|
24 |
+
return FastTextClassifierPipeline("model.bin")
|