istassiy commited on
Commit
f8aac2d
·
1 Parent(s): 2b0ca1b

commit from

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -50,10 +50,10 @@ def load_model():
50
  return model
51
 
52
  def sigmoid(x):
53
- return 1/(1 + np.exp(-x))
54
 
55
  def get_top_predictions(predictions):
56
- probs = (sigmoid(predictions) > 0.5).astype(float)
57
  probs = probs / np.sum(probs)
58
 
59
  res = {}
 
50
  return model
51
 
52
  def sigmoid(x):
53
+ return 1 / (1 + np.exp(-x))
54
 
55
  def get_top_predictions(predictions):
56
+ probs = sigmoid(predictions)
57
  probs = probs / np.sum(probs)
58
 
59
  res = {}