Update app.py

app.py

@@ -2,26 +2,16 @@ import gradio as gr
 import tensorflow as tf
 import numpy as np
 import librosa
-import requests
-import os
+from huggingface_hub import hf_hub_download
 
-# Download and load model
+# Load model with error handling
 try:
-    # Direct LFS URL for your model
-    model_url = "https://huggingface.co/IsolaHGVIS/Cat-Meow-Classification/resolve/main/best_model_fold_2.h5"
-    
-    # Create a models directory if it doesn't exist
-    os.makedirs('models', exist_ok=True)
-    model_path = 'models/best_model_fold_2.h5'
-    
-    # Download the model if it doesn't exist
-    if not os.path.exists(model_path):
-        print("Downloading model...")
-        response = requests.get(model_url)
-        with open(model_path, 'wb') as f:
-            f.write(response.content)
-        print("Model downloaded successfully!")
-    
+    # Download model file from Hugging Face with specific path
+    model_path = hf_hub_download(
+        repo_id="IsolaHGVIS/Cat-Meow-Classification",
+        filename="best_model_fold_2.h5",
+        repo_type="model"
+    )
     # Load the model
     model = tf.keras.models.load_model(model_path)
     print("Model loaded successfully!")