Update app.py

app.py

@@ -9,17 +9,16 @@ from streamlit_drawable_canvas import st_canvas
 from huggingface_hub import hf_hub_download
 from tensorflow import keras
 from huggingface_hub import from_pretrained_keras
-from urllib.request import urlretrieve
 
 
 # Load the model (using a Streamlit caching mechanism for efficiency)
 @st.cache(allow_output_mutation=True)  # Allow model mutation for prediction
 def load_model():
-    with tempfile.NamedTemporaryFile(suffix=".keras") as temp_file:
-        url = "https://huggingface.co/tomeheya/IM-417-128x128-classification/raw/main/IM_417_128.keras"
-        urlretrieve(url, temp_file.name)
-        return keras.models.load_model(temp_file.name) 
-    
+    files = [f for f in os.listdir('.') if os.path.isfile(f)]
+    print(f"FILES :: {files}")
+    return keras.models.load_model("IM_417_128.keras") 
+        
+        
 
 # Function to preprocess the input image
 def preprocess_image(image):