Update app.py

app.py

@@ -7,7 +7,6 @@ from huggingface_hub import from_pretrained_keras
 import numpy as np
 import gradio as gr
 
-characters = {'b', 'n', 'w', 'c', '3', '5', '8', '7', '6', 'x', 'e', 'm', '2', 'd', 'f', 'y', '4', 'g', 'p'}
 max_length = 5
 img_width = 200
 img_height = 50
@@ -18,14 +17,12 @@ prediction_model = keras.models.Model(
     model.get_layer(name="image").input, model.get_layer(name="dense2").output
 )
 
-# Mapping characters to integers
-char_to_num = layers.StringLookup(
-    vocabulary=list(characters), mask_token=None
-)
+with open("vocab.txt", "r") as f:
+    vocab = f.read().splitlines()
 
 # Mapping integers back to original characters
 num_to_char = layers.StringLookup(
-    vocabulary=char_to_num.get_vocabulary(), mask_token=None, invert=True
+    vocabulary=vocab, mask_token=None, invert=True
 )
 
 def decode_batch_predictions(pred):