YashuSinghal commited on
Commit
4c197a9
·
verified ·
1 Parent(s): 852b7cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,8 +13,8 @@ def predict(img):
13
  #img_array = tf.expand_dims(img_array, 0)
14
 
15
  predictions = model.predict(img_array)
16
- predicted_class = class_names[np.argmax(predictions[0])]
17
- confidence = round(100 * (np.max(predictions[0])), 2)
18
  return {class_names[i]: float(predictions[0][i]) for i in range(3)}
19
 
20
 
 
13
  #img_array = tf.expand_dims(img_array, 0)
14
 
15
  predictions = model.predict(img_array)
16
+ # predicted_class = class_names[np.argmax(predictions[0])]
17
+ # confidence = round(100 * (np.max(predictions[0])), 2)
18
  return {class_names[i]: float(predictions[0][i]) for i in range(3)}
19
 
20