update requirements

README.md

@@ -9,6 +9,7 @@ app_file: app.py
 pinned: false
 license: mit
 short_description: Fruits classifier using fastai and resnet
+python_version: 3.12
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -4,11 +4,10 @@ import gradio as gr
 
 learn = load_learner("fruits-model-v1.pkl")
 
-categories = ('Apple', 'Banana', 'Grape', 'Mango', 'Strawberry')
 
 def classify_fruit(img):
     pred, idx, probs = learn.predict(img)
-    return dict(zip(categories, map(float, probs)))
+    return {learn.dls.vocab[i]: float(probs[i]) for i in range(len(probs))}
 
 image = gr.Image(width=224, height=224, type="pil")
 label = gr.Label()

requirements-local.txt

@@ -0,0 +1,162 @@
+aiofiles==24.1.0
+aiohappyeyeballs==2.6.1
+aiohttp==3.13.3
+aiosignal==1.4.0
+annotated-doc==0.0.4
+annotated-types==0.7.0
+antlr4-python3-runtime==4.9.3
+anyio==4.12.1
+apsw==3.51.2.0
+apswutils==0.1.2
+asttokens==3.0.1
+attrs==25.4.0
+beartype==0.22.9
+beautifulsoup4==4.14.3
+blis==1.3.3
+blobfile==3.2.0
+brotli==1.2.0
+catalogue==2.0.10
+certifi==2026.2.25
+charset-normalizer==3.4.5
+click==8.3.1
+cloudpathlib==0.23.0
+cloudpickle==3.1.2
+confection==0.1.5
+contourpy==1.3.3
+cycler==0.12.1
+cymem==2.0.13
+datasets==4.6.1
+debugpy==1.8.20
+decorator==5.2.1
+dill==0.4.0
+executing==2.2.1
+fastai==2.8.7
+fastapi==0.135.1
+fastcore==1.12.18
+fastdownload==0.0.7
+fastlite==0.2.4
+fastprogress==1.1.5
+fasttransform==0.0.2
+ffmpy==1.0.0
+filelock==3.25.0
+fonttools==4.61.1
+frozenlist==1.8.0
+fsspec==2026.2.0
+gradio==5.50.0
+gradio_client==1.14.0
+groovy==0.1.2
+h11==0.16.0
+hf-xet==1.3.2
+httpcore==1.0.9
+httptools==0.7.1
+httpx==0.28.1
+huggingface_hub==1.6.0
+idna==3.11
+ipykernel==6.17.1
+ipython==9.11.0
+ipython_pygments_lexers==1.1.1
+itsdangerous==2.2.0
+jedi==0.19.2
+Jinja2==3.1.6
+joblib==1.5.3
+jupyter_client==8.8.0
+jupyter_core==5.9.1
+kagglehub==1.0.0
+kagglesdk==0.1.16
+kiwisolver==1.5.0
+lxml==6.0.2
+markdown-it-py==4.0.0
+MarkupSafe==3.0.3
+matplotlib==3.10.8
+matplotlib-inline==0.2.1
+mdurl==0.1.2
+mpmath==1.3.0
+multidict==6.7.1
+multiprocess==0.70.18
+murmurhash==1.0.15
+nest-asyncio==1.6.0
+networkx==3.6.1
+numpy==2.4.3
+oauthlib==3.3.1
+omegaconf==2.3.0
+orjson==3.11.7
+packaging==26.0
+pandas==2.3.3
+parso==0.8.6
+pexpect==4.9.0
+pillow==11.3.0
+platformdirs==4.9.4
+plum-dispatch==2.7.1
+preshed==3.0.12
+prompt_toolkit==3.0.52
+propcache==0.4.1
+protobuf==7.34.0
+psutil==7.2.2
+ptyprocess==0.7.0
+pure_eval==0.2.3
+pyarrow==23.0.1
+pycryptodomex==3.23.0
+pydantic==2.12.3
+pydantic_core==2.41.4
+pydub==0.25.1
+Pygments==2.19.2
+pyparsing==3.3.2
+python-dateutil==2.9.0.post0
+python-dotenv==1.2.2
+python-fasthtml==0.12.48
+python-multipart==0.0.22
+pytz==2026.1.post1
+PyYAML==6.0.3
+pyzmq==27.1.0
+regex==2026.2.28
+requests==2.32.5
+rich==14.3.3
+ruff==0.15.5
+safehttpx==0.1.7
+safetensors==0.7.0
+scikit-learn==1.8.0
+scipy==1.17.1
+semantic-version==2.10.0
+sentencepiece==0.2.1
+setuptools==82.0.1
+shellingham==1.5.4
+six==1.17.0
+smart_open==7.5.1
+soupsieve==2.8.3
+spacy==3.8.11
+spacy-legacy==3.0.12
+spacy-loggers==1.0.5
+srsly==2.5.2
+stack-data==0.6.3
+starlette==0.52.1
+sympy==1.14.0
+thinc==8.3.10
+threadpoolctl==3.6.0
+tiktoken==0.12.0
+tokenizers==0.22.2
+tomlkit==0.13.3
+torch @ https://download.pytorch.org/whl/cpu/torch-2.10.0%2Bcpu-cp312-cp312-manylinux_2_28_x86_64.whl#sha256=ee40b8a4b4b2cf0670c6fd4f35a7ef23871af956fecb238fbf5da15a72650b1d
+torchao==0.10.0
+torchdata==0.11.0
+torchsummary==1.5.1
+torchtune==0.6.1
+torchvision @ https://download.pytorch.org/whl/cpu/torchvision-0.25.0%2Bcpu-cp312-cp312-manylinux_2_28_x86_64.whl#sha256=c1be164e93c68b2dbf460fd58975377c892dbcf3358fb72941709c3857351bba
+tornado==6.5.4
+tqdm==4.67.3
+traitlets==5.14.3
+typer==0.24.1
+typer-slim==0.24.0
+typing-inspection==0.4.2
+typing_extensions==4.15.0
+tzdata==2025.3
+urllib3==2.6.3
+uvicorn==0.41.0
+uvloop==0.22.1
+wasabi==1.1.3
+watchfiles==1.1.1
+wcwidth==0.6.0
+weasel==0.4.3
+websockets==15.0.1
+wrapt==2.1.2
+xxhash==3.6.0
+yarl==1.23.0

requirements.txt

@@ -1,7 +1,162 @@
+aiofiles==24.1.0
+aiohappyeyeballs==2.6.1
+aiohttp==3.13.3
+aiosignal==1.4.0
+annotated-doc==0.0.4
+annotated-types==0.7.0
+antlr4-python3-runtime==4.9.3
+anyio==4.12.1
+apsw==3.51.2.0
+apswutils==0.1.2
+asttokens==3.0.1
+attrs==25.4.0
+beartype==0.22.9
+beautifulsoup4==4.14.3
+blis==1.3.3
+blobfile==3.2.0
+brotli==1.2.0
+catalogue==2.0.10
+certifi==2026.2.25
+charset-normalizer==3.4.5
+click==8.3.1
+cloudpathlib==0.23.0
+cloudpickle==3.1.2
+confection==0.1.5
+contourpy==1.3.3
+cycler==0.12.1
+cymem==2.0.13
+datasets==4.6.1
+debugpy==1.8.20
+decorator==5.2.1
+dill==0.4.0
+executing==2.2.1
 fastai==2.8.7
-fastcore==1.12.24
+fastapi==0.135.1
+fastcore==1.12.18
 fastdownload==0.0.7
 fastlite==0.2.4
 fastprogress==1.1.5
 fasttransform==0.0.2
-jupyter==1.1.1
+ffmpy==1.0.0
+filelock==3.25.0
+fonttools==4.61.1
+frozenlist==1.8.0
+fsspec==2026.2.0
+gradio==5.50.0
+gradio_client==1.14.0
+groovy==0.1.2
+h11==0.16.0
+hf-xet==1.3.2
+httpcore==1.0.9
+httptools==0.7.1
+httpx==0.28.1
+huggingface_hub==1.6.0
+idna==3.11
+ipykernel==6.17.1
+ipython==9.11.0
+ipython_pygments_lexers==1.1.1
+itsdangerous==2.2.0
+jedi==0.19.2
+Jinja2==3.1.6
+joblib==1.5.3
+jupyter_client==8.8.0
+jupyter_core==5.9.1
+kagglehub==1.0.0
+kagglesdk==0.1.16
+kiwisolver==1.5.0
+lxml==6.0.2
+markdown-it-py==4.0.0
+MarkupSafe==3.0.3
+matplotlib==3.10.8
+matplotlib-inline==0.2.1
+mdurl==0.1.2
+mpmath==1.3.0
+multidict==6.7.1
+multiprocess==0.70.18
+murmurhash==1.0.15
+nest-asyncio==1.6.0
+networkx==3.6.1
+numpy==2.4.3
+oauthlib==3.3.1
+omegaconf==2.3.0
+orjson==3.11.7
+packaging==26.0
+pandas==2.3.3
+parso==0.8.6
+pexpect==4.9.0
+pillow==11.3.0
+platformdirs==4.9.4
+plum-dispatch==2.7.1
+preshed==3.0.12
+prompt_toolkit==3.0.52
+propcache==0.4.1
+protobuf==7.34.0
+psutil==7.2.2
+ptyprocess==0.7.0
+pure_eval==0.2.3
+pyarrow==23.0.1
+pycryptodomex==3.23.0
+pydantic==2.12.3
+pydantic_core==2.41.4
+pydub==0.25.1
+Pygments==2.19.2
+pyparsing==3.3.2
+python-dateutil==2.9.0.post0
+python-dotenv==1.2.2
+python-fasthtml==0.12.48
+python-multipart==0.0.22
+pytz==2026.1.post1
+PyYAML==6.0.3
+pyzmq==27.1.0
+regex==2026.2.28
+requests==2.32.5
+rich==14.3.3
+ruff==0.15.5
+safehttpx==0.1.7
+safetensors==0.7.0
+scikit-learn==1.8.0
+scipy==1.17.1
+semantic-version==2.10.0
+sentencepiece==0.2.1
+setuptools==82.0.1
+shellingham==1.5.4
+six==1.17.0
+smart_open==7.5.1
+soupsieve==2.8.3
+spacy==3.8.11
+spacy-legacy==3.0.12
+spacy-loggers==1.0.5
+srsly==2.5.2
+stack-data==0.6.3
+starlette==0.52.1
+sympy==1.14.0
+thinc==8.3.10
+threadpoolctl==3.6.0
+tiktoken==0.12.0
+tokenizers==0.22.2
+tomlkit==0.13.3
+torch==2.10.0
+torchao==0.10.0
+torchdata==0.11.0
+torchsummary==1.5.1
+torchtune==0.6.1
+torchvision==0.25.0
+tornado==6.5.4
+tqdm==4.67.3
+traitlets==5.14.3
+typer==0.24.1
+typer-slim==0.24.0
+typing-inspection==0.4.2
+typing_extensions==4.15.0
+tzdata==2025.3
+urllib3==2.6.3
+uvicorn==0.41.0
+uvloop==0.22.1
+wasabi==1.1.3
+watchfiles==1.1.1
+wcwidth==0.6.0
+weasel==0.4.3
+websockets==15.0.1
+wrapt==2.1.2
+xxhash==3.6.0
+yarl==1.23.0