Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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('
|
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)}
|