Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import tensorflow as tf
 import librosa
 import numpy as np
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, from_pretrained_keras
 
 # Mel Spectrogram parameters
 n_fft = 512        # FFT window length
@@ -29,9 +29,10 @@ def CTCLoss(y_true, y_pred):
     return loss
 
 # Download model from Hugging Face Hub
-model_path = hf_hub_download(repo_id="kobrasoft/kobraspeech-rnn-cs", filename="saved_model.pb")
-with tf.keras.utils.custom_object_scope({'CTCLoss': CTCLoss}):
-    model = tf.keras.models.load_model(model_path)
+# model_path = hf_hub_download(repo_id="kobrasoft/kobraspeech-rnn-cs", filename="saved_model.pb")
+# with tf.keras.utils.custom_object_scope({'CTCLoss': CTCLoss}):
+#     model = tf.keras.models.load_model(model_path)
+model = from_pretrained_keras("kobrasoft/kobraspeech-rnn-cs")
 
 import pickle as pkl