data-silence commited on
Commit
264a654
1 Parent(s): 5d75a68

add config.json

Browse files
Files changed (2) hide show
  1. config.json +0 -5
  2. 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
- classifier = FastTextClassifierPipeline("fasttext_news_classifier.bin")
 
 
20
  return results
21
 
22
 
23
+ def pipeline(task="text-classification", model=None):
24
+ return FastTextClassifierPipeline("model.bin")