add everything

app.py

@@ -0,0 +1,47 @@
+import gradio as gr
+from datasets import load_dataset
+import numpy as np
+import torch
+import torch.nn as nn
+from torchvision.transforms import Compose, ColorJitter, ToTensor, RandomPerspective
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+
+
+with open("dogs_labels.txt", "r") as f:
+    labels = f.read().split('\n')
+
+
+num_labels = len(labels)
+id2label = {str(i): c for i, c in enumerate(labels)}
+
+
+def classify_image(inp):
+    # Load model
+    model = AutoModelForImageClassification.from_pretrained(pretrained_model_name_or_path="asusevski/vit-dog-classifier")
+
+    # Preprocess
+    model_preprocessor_name = "google/vit-base-patch16-224"
+    feature_extractor = AutoFeatureExtractor.from_pretrained(model_preprocessor_name)
+    augs = Compose(
+        [
+            ColorJitter(brightness=0.25, contrast=0.25, saturation=0.25, hue=0.4),
+            RandomPerspective(distortion_scale=0.3),
+            ToTensor(),
+        ]
+    )
+
+    inp = np.array(augs(inp.convert('RGB')))
+    
+    inp = torch.tensor(feature_extractor(images=inp)['pixel_values'])
+    preds = model(inp)['logits']
+    preds = torch.flatten(preds)
+    preds = nn.functional.softmax(preds, dim=0)
+    confidences = {labels[i]: preds[i].item() for i in range(num_labels)}
+    return confidences
+
+
+gr.Interface(fn=classify_image, 
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Label(num_top_classes=3),
+    examples=['./data/dogs/train/Yorkie/01.jpg', './data/dogs/train/Yorkie/02.jpg']
+).launch()

dogs_labels.txt

@@ -0,0 +1,71 @@
+Afghan
+African Wild Dog
+Airedale
+American  Spaniel
+American Hairless
+American Spaniel
+Basenji
+Basset
+Beagle
+Bearded Collie
+Bermaise
+Bichon Frise
+Blenheim
+Bloodhound
+Bluetick
+Border Collie
+Borzoi
+Boston Terrier
+Boxer
+Bull Mastiff
+Bull Terrier
+Bulldog
+Cairn
+Chihuahua
+Chinese Crested
+Chow
+Clumber
+Cockapoo
+Cocker
+Collie
+Corgi
+Coyote
+Dalmation
+Dhole
+Dingo
+Doberman
+Elk Hound
+French Bulldog
+German Sheperd
+Golden Retriever
+Great Dane
+Great Perenees
+Greyhound
+Groenendael
+Irish Spaniel
+Irish Wolfhound
+Japanese Spaniel
+Komondor
+Labradoodle
+Labrador
+Lhasa
+Malinois
+Maltese
+Mex Hairless
+Newfoundland
+Pekinese
+Pit Bull
+Pomeranian
+Poodle
+Pug
+Rhodesian
+Rottweiler
+Saint Bernard
+Schnauzer
+Scotch Terrier
+Shar_Pei
+Shiba Inu
+Shih-Tzu
+Siberian Husky
+Vizsla
+Yorkie

requirements.txt

@@ -0,0 +1,143 @@
+attrs==19.3.0
+Automat==0.8.0
+backcall==0.1.0
+bleach==3.1.1
+blinker==1.4
+blis==0.7.8
+catalogue==2.0.7
+certifi==2019.11.28
+cffi==1.15.0
+chardet==3.0.4
+charset-normalizer==2.0.12
+click==8.1.3
+cloud-init==20.1
+colorama==0.4.3
+command-not-found==0.3
+configobj==5.0.6
+constantly==15.1.0
+cryptography==37.0.2
+cymem==2.0.6
+dataclasses==0.6
+dbus-python==1.2.16
+decorator==4.4.2
+defusedxml==0.6.0
+distro==1.4.0
+distro-info===0.23ubuntu1
+entrypoints==0.3
+filelock==3.7.1
+html5lib==1.0.1
+httplib2==0.14.0
+huggingface-hub==0.8.1
+hyperlink==19.0.0
+idna==2.8
+importlib-metadata==1.5.0
+incremental==16.10.1
+ipykernel==5.2.0
+ipython==7.13.0
+ipython-genutils==0.2.0
+ipywidgets==6.0.0
+jedi==0.15.2
+Jinja2==2.10.1
+joblib==1.1.0
+jsonpatch==1.22
+jsonpointer==2.0
+jsonschema==3.2.0
+jupyter-client==6.1.2
+jupyter-console==6.0.0
+jupyter-core==4.6.3
+keyring==18.0.1
+langcodes==3.3.0
+language-selector==0.1
+launchpadlib==1.10.13
+lazr.restfulclient==0.14.2
+lazr.uri==1.0.3
+MarkupSafe==1.1.0
+mistune==0.8.4
+more-itertools==4.2.0
+murmurhash==1.0.7
+nbconvert==5.6.1
+nbformat==5.0.4
+netifaces==0.10.4
+nltk==3.7
+notebook==6.0.3
+numpy==1.23.0
+oauthlib==3.1.0
+packaging==21.3
+pandas==1.5.2
+pandocfilters==1.4.2
+parso==0.5.2
+pathy==0.6.2
+pdfminer.six==20220524
+pdfplumber==0.7.1
+pexpect==4.6.0
+pickleshare==0.7.5
+Pillow==9.1.1
+preshed==3.0.6
+prometheus-client==0.7.1
+prompt-toolkit==2.0.10
+pyasn1==0.4.2
+pyasn1-modules==0.2.1
+pycparser==2.21
+pydantic==1.8.2
+Pygments==2.3.1
+PyGObject==3.36.0
+PyHamcrest==1.9.0
+PyJWT==1.7.1
+pymacaroons==0.13.0
+PyNaCl==1.3.0
+pyOpenSSL==19.0.0
+pyparsing==3.0.9
+pyrsistent==0.15.5
+pyserial==3.4
+python-apt==2.0.0
+python-dateutil==2.8.2
+python-dbusmock==0.19
+python-debian===0.1.36ubuntu1
+pytz==2022.7.1
+PyYAML==5.3.1
+pyzmq==18.1.1
+regex==2022.6.2
+requests==2.22.0
+requests-unixsocket==0.2.0
+scikit-learn==1.1.1
+scipy==1.8.1
+SecretStorage==2.3.1
+Send2Trash==1.5.0
+sentence-transformers==2.2.2
+sentencepiece==0.1.96
+service-identity==18.1.0
+simplejson==3.16.0
+six==1.14.0
+smart-open==5.2.1
+spacy==3.3.1
+spacy-legacy==3.0.9
+spacy-loggers==1.0.2
+srsly==2.4.3
+ssh-import-id==5.10
+systemd-python==234
+terminado==0.8.2
+testpath==0.4.4
+thinc==8.0.17
+threadpoolctl==3.1.0
+tokenizers==0.12.1
+torch==1.12.0
+torchvision==0.13.0
+tornado==5.1.1
+tqdm==4.64.0
+traitlets==4.3.3
+transformers==4.20.1
+Twisted==18.9.0
+typer==0.4.2
+typing-extensions==4.3.0
+ubuntu-advantage-tools==20.3
+ufw==0.36
+unattended-upgrades==0.1
+urllib3==1.25.8
+wadllib==1.3.3
+Wand==0.6.7
+wasabi==0.9.1
+wcwidth==0.1.8
+webencodings==0.5.1
+widgetsnbextension==2.0.0
+zipp==1.0.0
+zope.interface==4.7.1