Gaetano Parente commited on
Commit
87fbdff
·
1 Parent(s): a9d51d0

add multi-classification model h5

Browse files
data/model/{model.keras → multi-classification.h5} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bfe8f18c11d7239d74d8b5147128156175ad33948e5a9a74e8c0cc0cd28d685b
3
- size 6833312
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d426f8dbb6afa55aa32b836887bd112e91c2475c5a88f8d9f425f18f79abcba
3
+ size 72329352
modules/binary-classification.keras DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a50abd61f0e541056d942290594338f10452c23ea2e26b960a5eba828bae1b64
3
- size 595304
 
 
 
 
modules/multilabel_classification.py CHANGED
@@ -84,7 +84,7 @@ def multi_classification(request):
84
  os.remove('temp.txt')
85
 
86
  def multi_classification_small(text):
87
- model = MODEL + 'multi-classification.keras'
88
  tokenizer = TOKEN + 'multi-classification-tokenizer.json'
89
  try:
90
  labels, max_label = predict(model, tokenizer, text)
 
84
  os.remove('temp.txt')
85
 
86
  def multi_classification_small(text):
87
+ model = MODEL + 'multi-classification.h5'
88
  tokenizer = TOKEN + 'multi-classification-tokenizer.json'
89
  try:
90
  labels, max_label = predict(model, tokenizer, text)