Update app.py

app.py

@@ -4,6 +4,10 @@ import tensorflow as tf
 
 model = tf.keras.models.load_model('model.h5')
 
+def softmax(x):
+    e_x = np.exp(x - np.max(x))
+    return e_x / e_x.sum()
+
 def temp(image):
     img = (image['composite'])
     img = np.expand_dims(img, axis=0).astype('float32')
@@ -11,7 +15,7 @@ def temp(image):
     y_pred = model.predict(img)
     prediction = {0:'অ', 1:'আ', 3:'ই', 4:'ঈ', 5:'উ', 6:'ঊ', 7:'ঋ', 8:'এ', 9:'ঐ', 10:'ও', 2:'ঔ'}
     for i in range(11):
-        if y_pred[0][i] == 1:
+        if np.argmax(softmax(y_pred[0])) == i:
             return prediction[i]
 
 iface = gr.Interface(