updated and imported model from hf hub

app.py

@@ -4,11 +4,12 @@ from tensorflow import keras
 import gradio as gr
 import numpy as np
 import cv2
+from huggingface_hub import hf_hub_download
 
 # Load the model
-model = tf.keras.models.load_model("image_classifier.keras")
+model_path = hf_hub_download(repo_id="jamesywu5/cifar10_model", filename="image_classifier.keras")
+model = tf.keras.models.load_model(model_path)
 
-# Define your class names manually (if you know them)
 class_names = ['Airplane', 'Car', 'Bird', 'Cat', 'Deer', 'Dog', 'Frog', 'Horse', 'Ship', 'Truck']
 
 def predict_fn(img):

requirements.txt

@@ -1,99 +1,6 @@
-absl-py==2.3.1
-aiofiles==24.1.0
-annotated-types==0.7.0
-anyio==4.9.0
-astunparse==1.6.3
-Brotli==1.1.0
-certifi==2025.7.14
-charset-normalizer==3.4.2
-click==8.2.1
-cloudpickle==3.1.1
-contourpy==1.3.2
-cycler==0.12.1
-fastapi==0.116.1
-ffmpy==0.6.0
-filelock==3.18.0
-flatbuffers==25.2.10
-fonttools==4.59.0
-fsspec==2025.7.0
-gast==0.4.0
-google-pasta==0.2.0
-gradio==5.37.0
-gradio_client==1.10.4
-groovy==0.1.2
-grpcio==1.73.1
-h11==0.16.0
-h5py==3.14.0
-hf-xet==1.1.5
-httpcore==1.0.9
-httpx==0.28.1
-huggingface-hub==0.33.4
-idna==3.10
-imageio==2.37.0
-Jinja2==3.1.6
-joblib==1.5.1
-keras==3.10.0
-kiwisolver==1.4.8
-lazy_loader==0.4
-libclang==18.1.1
-llvmlite==0.44.0
-Markdown==3.8.2
-markdown-it-py==3.0.0
-MarkupSafe==3.0.2
-matplotlib==3.10.3
-mdurl==0.1.2
-ml_dtypes==0.5.1
-namex==0.1.0
-networkx==3.4.2
-numba==0.61.2
-numpy==2.1.3
-opencv-python-headless==4.12.0.88
-opt_einsum==3.4.0
-optree==0.16.0
-orjson==3.11.0
-packaging==25.0
-pandas==2.3.1
-pillow==11.3.0
-protobuf==5.29.5
-pydantic==2.11.7
-pydantic_core==2.33.2
-pydub==0.25.1
-Pygments==2.19.2
-pyparsing==3.2.3
-python-dateutil==2.9.0.post0
-python-multipart==0.0.20
-pytz==2025.2
-PyYAML==6.0.2
-requests==2.32.4
-rich==14.0.0
-ruff==0.12.3
-safehttpx==0.1.6
-scikit-image==0.25.2
-scikit-learn==1.7.0
-scipy>=1.7.0,<1.13
-semantic-version>=2.10.0
-setuptools>=65.0.0
-shap>=0.41.0,<0.49
-shellingham>=1.3.0
-six>=1.16.0
-slicer>=0.0.7
-sniffio>=1.3.0
-starlette>=0.27.0
-tensorboard>=2.10.0
-tensorboard-data-server>=0.6.1
 tensorflow==2.10.1
-termcolor>=2.1.0
-threadpoolctl>=3.1.0
-tifffile>=2021.11.2
-tomlkit>=0.11.0
-tqdm>=4.62.0
-typer>=0.9.0
-typing-inspection>=0.4.0
-typing_extensions>=4.5.0
-tzdata>=2022.1
-urllib3>=1.26.0,<2.0
-uvicorn>=0.20.0
-websockets>=10.4
-Werkzeug>=2.2.0
-wheel>=0.38.0
-wrapt>=1.14.1
+gradio==5.37.0
+numpy
+opencv-python-headless
+scipy
+huggingface_hub