marcossalinas commited on
Commit
6df8d12
·
1 Parent(s): 75f9fa6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -61,6 +61,9 @@ def predict (image_path):
61
  for k in to_delete:
62
  del diseases[k]
63
 
 
 
 
64
 
65
  return (
66
  {'Enferma': float(result), 'Sana': 1 - float(result)}, diseases
 
61
  for k in to_delete:
62
  del diseases[k]
63
 
64
+ if len(diseases) == 0:
65
+ diseases = {'No specific disease': 0.0}
66
+
67
 
68
  return (
69
  {'Enferma': float(result), 'Sana': 1 - float(result)}, diseases