first commit

app.py

@@ -0,0 +1,28 @@
+import gradio as gr
+import os
+import torch
+from model import create_model
+from timeit import default_timer as Timer
+class_names=["Pizza","Steak","Sushi"]
+effnetb2,effnetb2_transform=create_model(num_class=3)
+effnetb2.load_state_dict(torch.load(f="demos\foodvision_mini\effnetb2_pizza_steak_sushi (1).pt",map_location=torch.device('cpu')))
+
+
+def predict(img):
+  start_time=timer()
+  img=effnetb2_transform(img).unsqueeze(0)
+  effnetb2.eval()
+  with torch.inference_mode():
+    predict_logit=effnetb2(img)
+    predict_prob=torch.softmax(predict_logit,dim=1)
+    predict_label=torch.argmax(predict_prob,dim=1)
+  pred_label_prob={class_names[i]:float(predict_prob[0][i]) for i in range(len(class_names))}
+  end_time=timer()
+  pred_time=round(end_time-start_time,4)
+  return pred_label_prob,pred_time
+
+title="Food Vision Mini 🍕🥩🍣"
+description="An EfficientNetB2 feature extractor computer vision model to classify images of food as pizza, steak or sushi."
+example_list=[['examples/'+example] for example in os.listdir("examples")]
+demo=gr.Interface(fn=predict,inputs=gr.Image(type="pil"),outputs=[gr.Label(num_top_classes=3,label="Prediction"),gr.Number(label="Prediction time")],examples=example_list,title=title,description=description)
+demo.launch()

effnetb2_pizza_steak_sushi (1).pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ee17107d6aa14ae47b961ab7b51e0a0a9c9ba04b5e35bf2fa546cca1d15677b5
+size 31284026

model.py

@@ -0,0 +1,13 @@
+import torch
+import torchvision
+from torchvision import transforms
+from torch import nn
+def create_model(num_class:int=3,seed:int=42):
+    weights=torchvision.models.EfficientNet_B2_Weights.DEFAULT
+    transform=weights.transforms()
+    model=torchvision.models.efficientnet_b2(weights=weights)
+    for params in model.parameters():
+        params.requires_grad=False
+    torch.manual_seed(seed)
+    model.classifier=nn.Sequential(nn.Dropout(p=0.3, inplace=True),nn.Linear(in_features=1408, out_features=num_class, bias=True))
+    return model,transform

requirements.txt

@@ -0,0 +1,369 @@
+absl-py==2.1.0
+aiobotocore==2.20.0
+aiofiles==24.1.0
+aiohappyeyeballs==2.4.6
+aiohttp==3.11.13
+aioitertools==0.12.0
+aiosignal==1.3.2
+altair==5.4.1
+anndata==0.11.3
+annotated-types==0.7.0
+anyio==4.6.2.post1
+argon2-cffi==23.1.0
+argon2-cffi-bindings==21.2.0
+array_api_compat==1.11
+arrow==1.3.0
+asciitree==0.3.3
+asttokens==2.4.1
+astunparse==1.6.3
+async-lru==2.0.4
+attrs==24.2.0
+babel==2.16.0
+backports.tarfile==1.2.0
+beautifulsoup4==4.12.3
+bleach==6.1.0
+blinker==1.8.2
+blis==1.2.0
+botocore==1.36.23
+cachetools==5.5.0
+catalogue==2.0.10
+certifi==2024.8.30
+cffi==1.17.1
+charset-normalizer==3.4.0
+click==8.1.7
+cloudpathlib==0.20.0
+cloudpickle==3.1.1
+colorama==0.4.6
+colorcet==3.1.0
+coloredlogs==15.0.1
+comm==0.2.2
+confection==0.1.5
+contourpy==1.3.1
+cycler==0.12.1
+cymem==2.0.11
+dask==2024.11.2
+dask-expr==1.1.19
+dask-image==2024.5.3
+dataclasses-json==0.6.7
+datashader==0.17.0
+debugpy==1.8.7
+decorator==5.1.1
+defusedxml==0.7.1
+Deprecated==1.2.18
+distributed==2024.11.2
+distro==1.9.0
+dlib==19.24.6
+docutils==0.21.2
+en_core_web_md @ https://github.com/explosion/spacy-models/releases/download/en_core_web_md-3.8.0/en_core_web_md-3.8.0-py3-none-any.whl#sha256=5e6329fe3fecedb1d1a02c3ea2172ee0fede6cea6e4aefb6a02d832dba78a310
+executing==2.1.0
+face-recognition==1.3.0
+face-recognition-models==0.3.0
+fastapi==0.115.12
+fasteners==0.19
+fastjsonschema==2.20.0
+ffmpy==0.5.0
+filelock==3.18.0
+Flask==3.0.3
+Flask-Cors==5.0.0
+flatbuffers==25.1.24
+fonttools==4.55.3
+fqdn==1.5.1
+frozenlist==1.5.0
+fsspec==2025.2.0
+gast==0.6.0
+geopandas==1.0.1
+gitdb==4.0.11
+GitPython==3.1.43
+google-ai-generativelanguage==0.6.15
+google-api-core==2.24.1
+google-api-python-client==2.162.0
+google-auth==2.38.0
+google-auth-httplib2==0.2.0
+google-generativeai==0.8.4
+google-pasta==0.2.0
+googleapis-common-protos==1.68.0
+gradio==5.26.0
+gradio_client==1.9.0
+greenlet==3.1.1
+groovy==0.1.2
+grpcio==1.71.0rc2
+grpcio-status==1.71.0rc2
+h11==0.14.0
+h5py==3.12.1
+httpcore==1.0.6
+httplib2==0.22.0
+httpx==0.27.2
+httpx-sse==0.4.0
+huggingface-hub==0.30.1
+humanfriendly==10.0
+id==1.5.0
+idna==3.10
+imageio==2.37.0
+imbalanced-learn==0.13.0
+imblearn==0.0
+importlib_metadata==8.6.1
+ipykernel==6.29.5
+ipython==8.28.0
+ipywidgets==8.1.5
+isoduration==20.11.0
+itsdangerous==2.2.0
+jaraco.classes==3.4.0
+jaraco.context==6.0.1
+jaraco.functools==4.1.0
+jax==0.5.0
+jaxlib==0.5.0
+jedi==0.19.1
+Jinja2==3.1.4
+jiter==0.9.0
+jmespath==1.0.1
+joblib==1.4.2
+json5==0.9.25
+jsonpatch==1.33
+jsonpointer==3.0.0
+jsonschema==4.23.0
+jsonschema-specifications==2024.10.1
+jupyter==1.1.1
+jupyter-console==6.6.3
+jupyter-events==0.10.0
+jupyter-lsp==2.2.5
+jupyter_client==8.6.3
+jupyter_core==5.7.2
+jupyter_server==2.14.2
+jupyter_server_terminals==0.5.3
+jupyterlab==4.2.5
+jupyterlab_pygments==0.3.0
+jupyterlab_server==2.27.3
+jupyterlab_widgets==3.0.13
+keras==3.8.0
+keyring==25.6.0
+kiwisolver==1.4.8
+langchain==0.3.23
+langchain-community==0.3.21
+langchain-core==0.3.51
+langchain-openai==0.3.12
+langchain-text-splitters==0.3.8
+langcodes==3.5.0
+langgraph==0.3.28
+langgraph-checkpoint==2.0.24
+langgraph-prebuilt==0.1.8
+langgraph-sdk==0.1.61
+langsmith==0.3.30
+language_data==1.3.0
+lazy_loader==0.4
+libclang==18.1.1
+lightning-utilities==0.14.2
+llvmlite==0.44.0
+locket==1.0.0
+marisa-trie==1.2.1
+Markdown==3.7
+markdown-it-py==3.0.0
+MarkupSafe==3.0.1
+marshmallow==3.26.1
+matplot==0.1.9
+matplotlib==3.10.0
+matplotlib-inline==0.1.7
+mdurl==0.1.2
+mediapipe==0.10.21
+mistune==3.0.2
+ml-dtypes==0.4.1
+monai==1.4.0
+more-itertools==10.7.0
+MouseInfo==0.1.3
+mpmath==1.3.0
+msgpack==1.1.0
+multidict==6.1.0
+multipledispatch==1.0.0
+multiscale_spatial_image==2.0.2
+murmurhash==1.0.12
+mypy-extensions==1.0.0
+mysql-connector-python==9.1.0
+namex==0.0.8
+narwhals==1.9.4
+natsort==8.4.0
+nbclient==0.10.0
+nbconvert==7.16.4
+nbformat==5.10.4
+nest-asyncio==1.6.0
+networkx==3.4.2
+nh3==0.2.21
+nltk==3.9.1
+notebook==7.2.2
+notebook_shim==0.2.4
+numba==0.61.0
+numcodecs==0.15.1
+numpy==1.26.4
+ome-zarr==0.10.3
+onnx==1.17.0
+onnxruntime==1.21.1
+onnxscript==0.2.4
+openai==1.72.0
+opencv-contrib-python==4.11.0.86
+opencv-python==4.11.0.86
+opt_einsum==3.4.0
+optree==0.14.0
+orjson==3.10.16
+ormsgpack==1.9.1
+overrides==7.7.0
+packaging==24.1
+pandas==2.2.3
+pandocfilters==1.5.1
+param==2.2.0
+parso==0.8.4
+partd==1.4.2
+pillow==10.4.0
+PIMS==0.7
+platformdirs==4.3.6
+pooch==1.8.2
+preshed==3.0.9
+prometheus_client==0.21.0
+prompt_toolkit==3.0.48
+propcache==0.3.0
+proto-plus==1.26.0
+protobuf==5.29.3
+psutil==6.1.0
+pure_eval==0.2.3
+pyarrow==17.0.0
+pyasn1==0.6.1
+pyasn1_modules==0.4.1
+PyAutoGUI==0.9.54
+pycparser==2.22
+pyct==0.5.0
+pydantic==2.10.6
+pydantic-settings==2.8.1
+pydantic_core==2.27.2
+pydeck==0.9.1
+pydub==0.25.1
+PyGetWindow==0.0.9
+Pygments==2.18.0
+pyloco==0.0.139
+PyMsgBox==1.0.9
+pynput==1.7.7
+pyogrio==0.10.0
+pyopencl==2025.1
+pyparsing==3.2.0
+pypdf==5.4.0
+pyperclip==1.9.0
+pyproj==3.7.1
+pyreadline3==3.5.4
+PyRect==0.2.0
+PyScreeze==1.0.1
+pytesseract==0.3.13
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.0
+python-json-logger==2.0.7
+python-multipart==0.0.20
+pytools==2025.1.2
+pytweening==1.2.0
+pytz==2024.2
+pywin32==308
+pywin32-ctypes==0.2.3
+pywinpty==2.0.14
+PyYAML==6.0.2
+pyzmq==26.2.0
+readme_renderer==44.0
+redis==5.1.1
+referencing==0.35.1
+regex==2024.9.11
+requests==2.32.3
+requests-toolbelt==1.0.0
+rfc3339-validator==0.1.4
+rfc3986==2.0.0
+rfc3986-validator==0.1.1
+rich==13.9.2
+rpds-py==0.20.0
+rsa==4.9
+ruff==0.11.7
+s3fs==2025.2.0
+safehttpx==0.1.6
+safetensors==0.5.3
+scikit-image==0.25.2
+scikit-learn==1.5.2
+scipy==1.14.1
+seaborn==0.13.2
+semantic-version==2.10.0
+Send2Trash==1.8.3
+sentencepiece==0.2.0
+shapely==2.0.7
+shellingham==1.5.4
+SimpleWebSocketServer==0.1.2
+six==1.16.0
+sklearn-compat==0.1.3
+slicerator==1.1.0
+smart-open==7.1.0
+smmap==5.0.1
+sniffio==1.3.1
+some_library==0.0.1
+sortedcontainers==2.4.0
+sounddevice==0.5.1
+soupsieve==2.6
+spacy==3.8.4
+spacy-legacy==3.0.12
+spacy-loggers==1.0.5
+spatial_image==1.2.1
+spatialdata==0.3.0
+spotipy==2.24.0
+SQLAlchemy==2.0.40
+srsly==2.5.1
+stack-data==0.6.3
+starlette==0.46.2
+streamlit==1.39.0
+sympy==1.13.1
+tblib==3.0.0
+tenacity==9.0.0
+tensorboard==2.18.0
+tensorboard-data-server==0.7.2
+tensorflow==2.18.0
+tensorflow-io-gcs-filesystem==0.31.0
+tensorflow_intel==2.18.0
+termcolor==2.5.0
+terminado==0.18.1
+thinc==8.3.4
+threadpoolctl==3.5.0
+tifffile==2025.2.18
+tiktoken==0.9.0
+tinycss2==1.3.0
+tokenizers==0.21.1
+toml==0.10.2
+tomlkit==0.13.2
+toolz==1.0.0
+torch==2.6.0
+torchinfo==1.8.0
+torchmetrics==1.7.0
+torchvision==0.21.0
+tornado==6.4.1
+tqdm==4.66.5
+traitlets==5.14.3
+transformers==4.50.3
+twine==6.1.0
+typer==0.15.2
+types-python-dateutil==2.9.0.20241003
+typing==3.7.4.3
+typing-inspect==0.9.0
+typing_extensions==4.12.2
+tzdata==2024.2
+uri-template==1.3.0
+uritemplate==4.1.1
+urllib3==2.2.3
+ushlex==0.99.1
+uvicorn==0.34.2
+wasabi==1.1.3
+watchdog==5.0.3
+wcwidth==0.2.13
+weasel==0.4.1
+webcolors==24.8.0
+webencodings==0.5.1
+websocket-client==1.8.0
+websockets==15.0.1
+Werkzeug==3.1.0
+widgetsnbextension==4.0.13
+wrapt==1.17.2
+xarray==2024.11.0
+xarray-dataclasses==1.9.1
+xarray-schema==0.0.3
+xarray-spatial==0.4.0
+xxhash==3.5.0
+yarl==1.18.3
+zarr==2.18.4
+zict==3.0.0
+zipp==3.21.0
+zstandard==0.23.0