dini15 commited on
Commit
a8f58d9
·
verified ·
1 Parent(s): 7676233

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ def run():
33
  predictions = model.predict(img_array)
34
 
35
  #Get class with the highest possibility
36
- idx = np.where(predictions => 0.5, 1, 0).item()
37
 
38
  type = ['oily', 'dry', 'normal']
39
  result = f'Prediction: {type[idx]}'
 
33
  predictions = model.predict(img_array)
34
 
35
  #Get class with the highest possibility
36
+ idx = np.where(predictions > 0.5, 1, 0).item()
37
 
38
  type = ['oily', 'dry', 'normal']
39
  result = f'Prediction: {type[idx]}'