Update app.py

app.py

@@ -1,11 +1,18 @@
 import gradio as gr
 import numpy as np
-from PIL import Image
 from tensorflow.keras.models import load_model
+from huggingface_hub import hf_hub_download
+from PIL import Image
+
+repo_id = "KevinJuanCarlos23/VehicleClassification" 
+filename = "vehicle_classification_model.keras" 
+
+model_path = hf_hub_download(repo_id=repo_id, filename=filename)
 
-model_path = "vehicle_classification_model.keras"  
+# Load model
 try:
     model = load_model(model_path)
+    print("Model loaded successfully!")
 except OSError as e:
     print("Error loading model:", e)