Update app.py

app.py

@@ -1,6 +1,8 @@
 import tensorflow as tf
 from tensorflow import keras
 from tensorflow.keras import layers
+from huggingface_hub import snapshot_download
+
 import numpy as np
 import gradio as gr
 
@@ -8,9 +10,11 @@ max_length = 5
 img_width = 200
 img_height = 50
 
+# Download the model from Hugging Face
+model_path = snapshot_download(repo_id="keras-io/ocr-for-captcha")
+
 # Load the TensorFlow SavedModel using TFSMLayer
-saved_model_path = "/home/user/.cache/huggingface/hub/models--keras-io--ocr-for-captcha/snapshots/1d695c4be3c72166292ff61c361d47c96f43cb7f"
-prediction_layer = keras.layers.TFSMLayer(saved_model_path, call_endpoint='serving_default')
+prediction_layer = keras.layers.TFSMLayer(model_path, call_endpoint='serving_default')
 
 with open("vocab.txt", "r") as f:
     vocab = f.read().splitlines()
@@ -60,4 +64,3 @@ iface = gr.Interface(classify_image, image, text,
 )
 
 iface.launch()
-