Spaces:
Sleeping
Sleeping
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:
|
| 3 |
-
size
|
|
|
|
| 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.
|
| 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)
|