Add application files for Gradio Space

app.py

@@ -0,0 +1,73 @@
+import gradio as gr
+import torch
+from torchvision import transforms
+import timm
+from PIL import Image
+import os
+
+# 1. Load Labels
+with open('labels.txt', 'r') as f:
+    labels = [line.strip() for line in f.readlines()]
+
+# 2. Model Definition
+def get_model(num_classes=200, model_path='models/final_model_best.pth'):
+    """Initializes and loads the pre-trained ConvNeXt V2 Large model."""
+    model = timm.create_model(
+        'convnextv2_large.fcmae_ft_in22k_in1k',
+        pretrained=False,
+        num_classes=num_classes,
+        drop_path_rate=0.2
+    )
+    if not os.path.exists(model_path):
+        print(f"Error: Model file not found at {model_path}.")
+        return None
+    try:
+        model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
+        model.eval()
+        print("Model loaded successfully.")
+        return model
+    except Exception as e:
+        print(f"An error occurred while loading the model: {e}")
+        return None
+
+model = get_model()
+
+# 3. Image Transformations
+transform = transforms.Compose([
+    transforms.Resize((224, 224)),
+    transforms.ToTensor(),
+    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
+])
+
+# 4. Prediction Function
+def predict(image):
+    """Takes a PIL image and returns a dictionary of top 3 predictions."""
+    if model is None:
+        return {"Error": "Model is not loaded. Please check the logs for errors."}
+        
+    image = transform(image).unsqueeze(0)
+    with torch.no_grad():
+        outputs = model(image)
+        probabilities = torch.nn.functional.softmax(outputs, dim=1)[0]
+        
+    # Get top 3 predictions
+    top3_prob, top3_indices = torch.topk(probabilities, 3)
+    
+    confidences = {labels[i]: float(p) for i, p in zip(top3_indices, top3_prob)}
+    
+    return confidences
+
+# 5. Gradio Interface
+title = "Bird Species Classifier"
+description = "Upload an image of a bird to classify it into one of 200 species. This model is a ConvNeXt V2 Large, fine-tuned on a dataset of 200 bird species."
+
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil", label="Upload Bird Image"),
+    outputs=gr.Label(num_top_classes=3, label="Predictions"),
+    title=title,
+    description=description,
+)
+
+if __name__ == "__main__":
+    iface.launch()

labels.txt

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

models/final_model_best.pth

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

requirements.txt

@@ -0,0 +1,7 @@
+torch
+torchvision
+gradio
+timm
+numpy
+tqdm
+Pillow