final on the main branch

app.py

@@ -0,0 +1,13 @@
+import gradio as gr
+from fastai.vision.all import *
+
+categories = ["great white", "zebra", "whale", "lemon", "thresher"]
+learner = load_learner("shark_recognizer.pkl")
+def recognize_shark(img):
+    pred, idx, probs = learner.predict(img)
+    return dict(zip(categories, map(float, probs)))
+    
+image = gr.inputs.Image()
+label = gr.outputs.Label()
+iface = gr.Interface(fn=recognize_shark, inputs=image, outputs=label)
+iface.launch()

requirements.txt

@@ -0,0 +1,117 @@
+aiofiles==23.1.0
+aiohttp==3.8.5
+aiosignal==1.3.1
+altair==5.0.1
+annotated-types==0.5.0
+anyio==3.7.1
+async-timeout==4.0.2
+attrs==23.1.0
+blis==0.7.9
+Brotli==1.0.9
+catalogue==2.0.8
+certifi==2023.5.7
+charset-normalizer==3.2.0
+click==8.1.6
+cmake==3.27.0
+confection==0.1.0
+contourpy==1.1.0
+cycler==0.11.0
+cymem==2.0.7
+duckduckgo-search==3.8.4
+exceptiongroup==1.1.2
+fastai==2.7.12
+fastapi==0.100.0
+fastcore==1.5.29
+fastdownload==0.0.7
+fastprogress==1.0.3
+ffmpy==0.3.1
+filelock==3.12.2
+fonttools==4.41.0
+frozenlist==1.4.0
+fsspec==2023.6.0
+gradio==3.37.0
+gradio_client==0.2.10
+h11==0.14.0
+h2==4.1.0
+hpack==4.0.0
+httpcore==0.17.3
+httpx==0.24.1
+huggingface-hub==0.16.4
+hyperframe==6.0.1
+idna==3.4
+Jinja2==3.1.2
+joblib==1.3.1
+jsonschema==4.18.4
+jsonschema-specifications==2023.7.1
+kiwisolver==1.4.4
+langcodes==3.3.0
+linkify-it-py==2.0.2
+lit==16.0.6
+lxml==4.9.3
+markdown-it-py==2.2.0
+MarkupSafe==2.1.3
+matplotlib==3.7.2
+mdit-py-plugins==0.3.3
+mdurl==0.1.2
+mpmath==1.3.0
+multidict==6.0.4
+murmurhash==1.0.9
+networkx==3.1
+numpy==1.25.1
+nvidia-cublas-cu11==11.10.3.66
+nvidia-cuda-cupti-cu11==11.7.101
+nvidia-cuda-nvrtc-cu11==11.7.99
+nvidia-cuda-runtime-cu11==11.7.99
+nvidia-cudnn-cu11==8.5.0.96
+nvidia-cufft-cu11==10.9.0.58
+nvidia-curand-cu11==10.2.10.91
+nvidia-cusolver-cu11==11.4.0.1
+nvidia-cusparse-cu11==11.7.4.91
+nvidia-nccl-cu11==2.14.3
+nvidia-nvtx-cu11==11.7.91
+orjson==3.9.2
+packaging==23.1
+pandas==2.0.3
+pathy==0.10.2
+Pillow==10.0.0
+preshed==3.0.8
+pydantic==1.10.11
+pydantic_core==2.3.0
+pydub==0.25.1
+pyparsing==3.0.9
+python-dateutil==2.8.2
+python-multipart==0.0.6
+pytz==2023.3
+PyYAML==6.0.1
+referencing==0.30.0
+requests==2.31.0
+rpds-py==0.9.2
+scikit-learn==1.3.0
+scipy==1.11.1
+semantic-version==2.10.0
+six==1.16.0
+smart-open==6.3.0
+sniffio==1.3.0
+socksio==1.0.0
+spacy==3.6.0
+spacy-legacy==3.0.12
+spacy-loggers==1.0.4
+srsly==2.4.7
+starlette==0.27.0
+sympy==1.12
+thinc==8.1.10
+threadpoolctl==3.2.0
+toolz==0.12.0
+torch==2.0.1
+torchvision==0.15.2
+tqdm==4.65.0
+triton==2.0.0
+typer==0.9.0
+typing_extensions==4.7.1
+tzdata==2023.3
+uc-micro-py==1.0.2
+urllib3==2.0.4
+uvicorn==0.23.1
+wasabi==1.1.2
+websockets==11.0.3
+yarl==1.9.2

shark_recognizer.pkl

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