Update app.py

app.py

@@ -1,3 +1,22 @@
+import streamlit as st
+from tensorflow import keras
+import numpy as np
+from huggingface_hub import HfFileSystem
+from PIL import Image
+
+# Authenticate and download the custom model from Hugging Face Spaces
+fs = HfFileSystem()
+model_path = 'dhhd255/main_model/best_model.h5'
+with fs.open(model_path, 'rb') as f:
+    model_content = f.read()
+
+# Save the model file to disk
+with open('best_model.h5', 'wb') as f:
+    f.write(model_content)
+
+# Load your custom model
+model = keras.models.load_model('best_model.h5')
+
 # Define a function that takes an image as input and uses the model for inference
 def image_classifier(image):
     # Preprocess the input image