fix

data/model/medical-image-classification.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:179db837e27060ab936729a443a0eb8e672edf2230bc1166137a0231e88c25ee
+size 134089704

modules/image_classification.py

@@ -24,7 +24,7 @@ def image_predict(image_path, model_path):
     return predictions, pred_labels
 
 def image_classification(numpy_image):
-    model = MODEL + 'medical-image-classification.keras' 
+    model = MODEL + 'medical-image-classification.h5' 
     myuuid = uuid.uuid4()
     filename = 'test_' + str(myuuid) + '.png'
     cv2.imwrite(filename, numpy_image)

modules/multilabel_classification.py

@@ -42,11 +42,11 @@ def multi_classification(text):
     tokenizer = TOKEN + 'multi-classification-tokenizer.json'
     myuuid = uuid.uuid4()
     filepath = BASE_PATH + 'temp' + str(myuuid) + '.txt'
-    try:
-        labels, max_label = predict(model, tokenizer, text, filepath)
-        response = {}
-        for i, label in enumerate(labels[0]):
-            response[class_names[i]] = "%.4f" % float(label)
-        return response
-    finally:
-        os.remove(filepath)
+    #try:
+    labels, max_label = predict(model, tokenizer, text, filepath)
+    response = {}
+    for i, label in enumerate(labels[0]):
+        response[class_names[i]] = "%.4f" % float(label)
+    return response
+    #finally:
+    #    os.remove(filepath)