Update app.py

app.py

@@ -1,5 +1,6 @@
 import subprocess
 import sys
+import os
 
 # Install TensorFlow
 subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'tensorflow'])
@@ -10,8 +11,11 @@ from tensorflow.keras.preprocessing.sequence import pad_sequences
 from tensorflow.keras.preprocessing.text import Tokenizer
 import gradio as gr
 
+# Define the path to the model file in the /mnt/data/ directory
+model_path = '/mnt/data/models/Bajiyo/Named_entity_transliteration_malayalam/best_model.h5'
+
 # Load your custom Keras model
-model = load_model('models/Bajiyo/Named_entity_transliteration_malayalam/best_model.h5')
+model = load_model(model_path)
 tokenizer = source_tokenizer
 
 # Function for transliteration