Amirah commited on
Commit
2bde588
1 Parent(s): a73b649

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ import gradio as gr
16
  cat = ['EIGHT', 'FIVE', 'FOUR', 'NINE', 'ONE', 'SEVEN', 'SIX', 'THREE', 'TWO', 'ZERO']
17
 
18
  def predict_image(img):
19
- model=tf.keras.models.load_model('Exp33DIG.hdf5')
20
  img_4d=img.reshape(-1,224,224,1)
21
  prediction=model.predict(img_4d)[0]
22
  return {cat[i]: float(prediction[i]) for i in range(10)}
 
16
  cat = ['EIGHT', 'FIVE', 'FOUR', 'NINE', 'ONE', 'SEVEN', 'SIX', 'THREE', 'TWO', 'ZERO']
17
 
18
  def predict_image(img):
19
+ model=tf.keras.models.load_model('Exp32DIG.hdf5')
20
  img_4d=img.reshape(-1,224,224,1)
21
  prediction=model.predict(img_4d)[0]
22
  return {cat[i]: float(prediction[i]) for i in range(10)}