sguna commited on
Commit
c46febd
·
verified ·
1 Parent(s): 30e75f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -44,6 +44,7 @@ def classify_image(image):
44
 
45
  # Return raw probabilities
46
  return {
 
47
  "safe": safe_prob, # Leave as a fraction (e.g., 0.92)
48
  "unsafe": unsafe_prob # Leave as a fraction (e.g., 0.08)
49
  }
 
44
 
45
  # Return raw probabilities
46
  return {
47
+ "unknown" : 0.2,
48
  "safe": safe_prob, # Leave as a fraction (e.g., 0.92)
49
  "unsafe": unsafe_prob # Leave as a fraction (e.g., 0.08)
50
  }