Upload folder using huggingface_hub

.github/workflows/update_space.yml

@@ -0,0 +1,28 @@
+name: Run Python script
+
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+  build:
+    runs-on: ubuntu-latest
+
+    steps:
+    - name: Checkout
+      uses: actions/checkout@v2
+
+    - name: Set up Python
+      uses: actions/setup-python@v2
+      with:
+        python-version: '3.9'
+
+    - name: Install Gradio
+      run: python -m pip install gradio
+
+    - name: Log in to Hugging Face
+      run: python -c 'import huggingface_hub; huggingface_hub.login(token="${{ secrets.hf_token }}")'
+
+    - name: Deploy to Spaces
+      run: gradio deploy

README.md

@@ -1,12 +1,6 @@
 ---
-title: Food Image Classifier
-emoji: 🐠
-colorFrom: indigo
-colorTo: blue
-sdk: gradio
-sdk_version: 5.8.0
+title: food-image-classifier
 app_file: app.py
-pinned: false
+sdk: gradio
+sdk_version: 5.7.1
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,72 @@
+import gradio as gr
+import torch
+import torchvision.transforms as transforms
+from PIL import Image
+import os
+from pathlib import Path
+
+class FoodImageClassifier:
+    def __init__(self, model_dir="traced_models/food_101_vit_small",
+                     model_file_name="model.pt",
+                     labels_path='food_101_classes.txt'):
+        self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+        print(self.device)
+        # Load the traced model
+        model_full_path = Path(model_dir,model_file_name)
+        self.model = torch.jit.load(model_full_path)
+        self.model = self.model.to(self.device)
+        self.model.eval()
+
+        # Define the same transforms used during training/testing
+        self.transforms = transforms.Compose([
+            transforms.Resize(256),
+            transforms.CenterCrop(224),
+            transforms.ToTensor(),
+            transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
+        ])
+
+        # Load labels from file
+        with open(labels_path, 'r') as f:
+            self.labels = [line.strip() for line in f.readlines()]
+        
+    @torch.no_grad()
+    def predict(self, image):
+        if image is None:
+            return None
+        
+        # Convert to PIL Image if needed
+        if not isinstance(image, Image.Image):
+            image = Image.fromarray(image).convert('RGB')
+        
+        # Preprocess image
+        img_tensor = self.transforms(image).unsqueeze(0).to(self.device)
+        
+        # Get prediction
+        output = self.model(img_tensor)
+        probabilities = torch.nn.functional.softmax(output[0], dim=0)
+        
+        # Create prediction dictionary
+        return {
+            self.labels[idx]: float(prob)
+            for idx, prob in enumerate(probabilities)
+        }
+
+# Create classifier instance
+classifier = FoodImageClassifier()
+
+# Create Gradio interface
+demo = gr.Interface(
+    fn=classifier.predict,
+    inputs=gr.Image(),
+    outputs=gr.Label(num_top_classes=5),
+    title="Food classifier",
+    description="Upload an image to classify Food Images",
+    examples=[
+        ["sample_data/apple_pie.jpg"],
+        ["sample_data/pizza.jpg"]
+    ]
+)
+
+
+if __name__ == "__main__":
+    demo.launch()

bird_200_classes.txt

@@ -0,0 +1,200 @@
+Acadian_Flycatcher
+American_Crow
+American_Goldfinch
+American_Pipit
+American_Redstart
+American_Three_toed_Woodpecker
+Anna_Hummingbird
+Artic_Tern
+Baird_Sparrow
+Baltimore_Oriole
+Bank_Swallow
+Barn_Swallow
+Bay_breasted_Warbler
+Belted_Kingfisher
+Bewick_Wren
+Black_Tern
+Black_and_white_Warbler
+Black_billed_Cuckoo
+Black_capped_Vireo
+Black_footed_Albatross
+Black_throated_Blue_Warbler
+Black_throated_Sparrow
+Blue_Grosbeak
+Blue_Jay
+Blue_headed_Vireo
+Blue_winged_Warbler
+Boat_tailed_Grackle
+Bobolink
+Bohemian_Waxwing
+Brandt_Cormorant
+Brewer_Blackbird
+Brewer_Sparrow
+Bronzed_Cowbird
+Brown_Creeper
+Brown_Pelican
+Brown_Thrasher
+Cactus_Wren
+California_Gull
+Canada_Warbler
+Cape_Glossy_Starling
+Cape_May_Warbler
+Cardinal
+Carolina_Wren
+Caspian_Tern
+Cedar_Waxwing
+Cerulean_Warbler
+Chestnut_sided_Warbler
+Chipping_Sparrow
+Chuck_will_Widow
+Clark_Nutcracker
+Clay_colored_Sparrow
+Cliff_Swallow
+Common_Raven
+Common_Tern
+Common_Yellowthroat
+Crested_Auklet
+Dark_eyed_Junco
+Downy_Woodpecker
+Eared_Grebe
+Eastern_Towhee
+Elegant_Tern
+European_Goldfinch
+Evening_Grosbeak
+Field_Sparrow
+Fish_Crow
+Florida_Jay
+Forsters_Tern
+Fox_Sparrow
+Frigatebird
+Gadwall
+Geococcyx
+Glaucous_winged_Gull
+Golden_winged_Warbler
+Grasshopper_Sparrow
+Gray_Catbird
+Gray_Kingbird
+Gray_crowned_Rosy_Finch
+Great_Crested_Flycatcher
+Great_Grey_Shrike
+Green_Jay
+Green_Kingfisher
+Green_Violetear
+Green_tailed_Towhee
+Groove_billed_Ani
+Harris_Sparrow
+Heermann_Gull
+Henslow_Sparrow
+Herring_Gull
+Hooded_Merganser
+Hooded_Oriole
+Hooded_Warbler
+Horned_Grebe
+Horned_Lark
+Horned_Puffin
+House_Sparrow
+House_Wren
+Indigo_Bunting
+Ivory_Gull
+Kentucky_Warbler
+Laysan_Albatross
+Lazuli_Bunting
+Le_Conte_Sparrow
+Least_Auklet
+Least_Flycatcher
+Least_Tern
+Lincoln_Sparrow
+Loggerhead_Shrike
+Long_tailed_Jaeger
+Louisiana_Waterthrush
+Magnolia_Warbler
+Mallard
+Mangrove_Cuckoo
+Marsh_Wren
+Mockingbird
+Mourning_Warbler
+Myrtle_Warbler
+Nashville_Warbler
+Nelson_Sharp_tailed_Sparrow
+Nighthawk
+Northern_Flicker
+Northern_Fulmar
+Northern_Waterthrush
+Olive_sided_Flycatcher
+Orange_crowned_Warbler
+Orchard_Oriole
+Ovenbird
+Pacific_Loon
+Painted_Bunting
+Palm_Warbler
+Parakeet_Auklet
+Pelagic_Cormorant
+Philadelphia_Vireo
+Pied_Kingfisher
+Pied_billed_Grebe
+Pigeon_Guillemot
+Pileated_Woodpecker
+Pine_Grosbeak
+Pine_Warbler
+Pomarine_Jaeger
+Prairie_Warbler
+Prothonotary_Warbler
+Purple_Finch
+Red_bellied_Woodpecker
+Red_breasted_Merganser
+Red_cockaded_Woodpecker
+Red_eyed_Vireo
+Red_faced_Cormorant
+Red_headed_Woodpecker
+Red_legged_Kittiwake
+Red_winged_Blackbird
+Rhinoceros_Auklet
+Ring_billed_Gull
+Ringed_Kingfisher
+Rock_Wren
+Rose_breasted_Grosbeak
+Ruby_throated_Hummingbird
+Rufous_Hummingbird
+Rusty_Blackbird
+Sage_Thrasher
+Savannah_Sparrow
+Sayornis
+Scarlet_Tanager
+Scissor_tailed_Flycatcher
+Scott_Oriole
+Seaside_Sparrow
+Shiny_Cowbird
+Slaty_backed_Gull
+Song_Sparrow
+Sooty_Albatross
+Spotted_Catbird
+Summer_Tanager
+Swainson_Warbler
+Tennessee_Warbler
+Tree_Sparrow
+Tree_Swallow
+Tropical_Kingbird
+Vermilion_Flycatcher
+Vesper_Sparrow
+Warbling_Vireo
+Western_Grebe
+Western_Gull
+Western_Meadowlark
+Western_Wood_Pewee
+Whip_poor_Will
+White_Pelican
+White_breasted_Kingfisher
+White_breasted_Nuthatch
+White_crowned_Sparrow
+White_eyed_Vireo
+White_necked_Raven
+White_throated_Sparrow
+Wilson_Warbler
+Winter_Wren
+Worm_eating_Warbler
+Yellow_Warbler
+Yellow_bellied_Flycatcher
+Yellow_billed_Cuckoo
+Yellow_breasted_Chat
+Yellow_headed_Blackbird
+Yellow_throated_Vireo

certificate.pem

@@ -0,0 +1,31 @@
+-----BEGIN CERTIFICATE-----
+MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
+TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
+cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
+WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
+ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
+MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
+h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
+0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
+A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
+T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
+B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
+B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
+KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
+OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
+jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
+qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
+rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
+HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
+hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
+ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
+3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
+NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
+ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
+TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
+jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
+oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
+4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
+mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
+emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
+-----END CERTIFICATE-----

flagged/dataset1.csv

@@ -0,0 +1,3 @@
+name,output,timestamp
+Chirag,Hello Chirag!,2024-12-03 00:12:21.409022
+Mitesh!,Hello Mitesh!!,2024-12-03 00:12:34.555481

flagged/dataset2.csv

@@ -0,0 +1,2 @@
+name,checkbox,value,output,timestamp
+HowDy Rahul,true,85,"name='HowDy Rahul', checkbox=True, value=85",2024-12-03 00:19:02.062029

flagged/dataset3.csv

@@ -0,0 +1,3 @@
+input_img,output,timestamp
+,.gradio/flagged/output/2c0088c869f226b53710/image.webp,2024-12-03 00:23:43.040231
+.gradio/flagged/input_img/596af0993c7fe8e1a18d/Screenshot 2024-11-28 at 1.03.22PM.png,.gradio/flagged/output/a13ab597cd4b1c6746c1/image.webp,2024-12-03 00:23:55.988759

food_101_classes.txt

@@ -0,0 +1,101 @@
+apple_pie
+baby_back_ribs
+baklava
+beef_carpaccio
+beef_tartare
+beet_salad
+beignets
+bibimbap
+bread_pudding
+breakfast_burrito
+bruschetta
+caesar_salad
+cannoli
+caprese_salad
+carrot_cake
+ceviche
+cheese_plate
+cheesecake
+chicken_curry
+chicken_quesadilla
+chicken_wings
+chocolate_cake
+chocolate_mousse
+churros
+clam_chowder
+club_sandwich
+crab_cakes
+creme_brulee
+croque_madame
+cup_cakes
+deviled_eggs
+donuts
+dumplings
+edamame
+eggs_benedict
+escargots
+falafel
+filet_mignon
+fish_and_chips
+foie_gras
+french_fries
+french_onion_soup
+french_toast
+fried_calamari
+fried_rice
+frozen_yogurt
+garlic_bread
+gnocchi
+greek_salad
+grilled_cheese_sandwich
+grilled_salmon
+guacamole
+gyoza
+hamburger
+hot_and_sour_soup
+hot_dog
+huevos_rancheros
+hummus
+ice_cream
+lasagna
+lobster_bisque
+lobster_roll_sandwich
+macaroni_and_cheese
+macarons
+miso_soup
+mussels
+nachos
+omelette
+onion_rings
+oysters
+pad_thai
+paella
+pancakes
+panna_cotta
+peking_duck
+pho
+pizza
+pork_chop
+poutine
+prime_rib
+pulled_pork_sandwich
+ramen
+ravioli
+red_velvet_cake
+risotto
+samosa
+sashimi
+scallops
+seaweed_salad
+shrimp_and_grits
+spaghetti_bolognese
+spaghetti_carbonara
+spring_rolls
+steak
+strawberry_shortcake
+sushi
+tacos
+takoyaki
+tiramisu
+tuna_tartare
+waffles

requirements.txt

@@ -0,0 +1,3 @@
+torch
+gradio
+torchvision

traced_models/bird_200_mobilenetv3_large_optimized/model.pt

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

traced_models/food_101_vit_small/model.pt

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

traced_models/food_101_vit_tiny/model.pt

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