Update app.py

app.py

@@ -31,7 +31,8 @@ def simple_normalizer(X):
 
 def predict (image_path):
   image = simple_normalizer(cut_and_resize(cv2.imread(image_path)))
-  result = predictor_disease_risk.predict(np.array([image]))[0][0]
+  result = predictor_disease_risk.predict(np.array([image]))[0][0] 
+  
   dr = predictor_dr.predict(np.array([image]))[0][0]  
   mh = predictor_mh.predict(np.array([image]))[0][0]
   odc = predictor_odc.predict(np.array([image]))[0][0] 
@@ -41,17 +42,23 @@ def predict (image_path):
   mya = predictor_mya.predict(np.array([image]))[0][0]
   brvo = predictor_brvo.predict(np.array([image]))[0][0]
   
-  return (
-    {'Enferma': float(result), 'Sana': 1 - float(result)},
-    {
+  diseases = {
       'DR'  : float(dr),
       'MH'  : float(mh),
       'ODC' : float(odc),
       'TSLN': float(tsln),
       'ARMD': float(armd),
       'MYA' : float(mya),
-      'BRVO': float(brvo)
-    }
+      'BRVO': float(brvo)    
+  }
+  
+  for k,v in diseases.items():
+    if v < 0.01:
+      del diseases[k]    
+  
+  
+  return (
+    {'Enferma': float(result), 'Sana': 1 - float(result)}, diseases
   )
  
 title = 'RetinAI (versión alfa)'