added main app and mapping

CarLabelToNameMapping.txt

@@ -0,0 +1,179 @@
+0  AM General Hummer SUV 2000
+1  Acura RL Sedan 2012
+2  Acura TL Sedan 2012
+3  Acura TL Type-S 2008
+4  Acura TSX Sedan 2012
+5  Acura Integra Type R 2001
+6  Acura ZDX Hatchback 2012
+7  Aston Martin V8 Vantage Convertible 2012
+8  Aston Martin V8 Vantage Coupe 2012
+9  Aston Martin Virage Convertible 2012
+10 Aston Martin Virage Coupe 2012
+11 Audi RS 4 Convertible 2008
+12 Audi A5 Coupe 2012
+13 Audi TTS Coupe 2012
+14 Audi R8 Coupe 2012
+15 Audi V8 Sedan 1994
+16 Audi 100 Sedan 1994
+17 Audi 100 Wagon 1994
+18 Audi TT Hatchback 2011
+19 Audi S6 Sedan 2011
+20 Audi S5 Convertible 2012
+21 Audi S5 Coupe 2012
+22 Audi S4 Sedan 2012
+23 Audi S4 Sedan 2007
+24 Audi TT RS Coupe 2012
+25 BMW ActiveHybrid 5 Sedan 2012
+26 BMW 1 Series Convertible 2012
+27 BMW 1 Series Coupe 2012
+28 BMW 3 Series Sedan 2012
+29 BMW 3 Series Wagon 2012
+30 BMW 6 Series Convertible 2007
+31 BMW X5 SUV 2007
+32 BMW X6 SUV 2012
+35 BMW M6 Convertible 2010
+36 BMW X3 SUV 2012
+37 BMW Z4 Convertible 2012
+38 Bentley Continental Supersports Conv. Convertible 2012
+39 Bentley Arnage Sedan 2009
+40 Bentley Mulsanne Sedan 2011
+41 Bentley Continental GT Coupe 2012
+42 Bentley Continental GT Coupe 2007
+43 Bentley Continental Flying Spur Sedan 2007
+44 Bugatti Veyron 16.4 Convertible 2009
+45 Bugatti Veyron 16.4 Coupe 2009
+46 Buick Regal GS 2012
+47 Buick Rainier SUV 2007
+48 Buick Verano Sedan 2012
+49 Buick Enclave SUV 2012
+50 Cadillac CTS-V Sedan 2012
+52 Cadillac Escalade EXT Crew Cab 2007
+53 Chevrolet Silverado 1500 Hybrid Crew Cab 2012
+54 Chevrolet Corvette Convertible 2012
+55 Chevrolet Corvette ZR1 2012
+56 Chevrolet Corvette Ron Fellows Edition Z06 2007
+57 Chevrolet Traverse SUV 2012
+58 Chevrolet Camaro Convertible 2012
+59 Chevrolet HHR SS 2010
+60 Chevrolet Impala Sedan 2007
+61 Chevrolet Tahoe Hybrid SUV 2012
+62 Chevrolet Sonic Sedan 2012
+63 Chevrolet Express Cargo Van 2007
+64 Chevrolet Avalanche Crew Cab 2012
+65 Chevrolet Cobalt SS 2010
+66 Chevrolet Malibu Hybrid Sedan 2010
+67 Chevrolet TrailBlazer SS 2009
+68 Chevrolet Silverado 2500HD Regular Cab 2012
+69 Chevrolet Silverado 1500 Classic Extended Cab 2007
+70 Chevrolet Express Van 2007
+71 Chevrolet Monte Carlo Coupe 2007
+72 Chevrolet Malibu Sedan 2007
+73 Chevrolet Silverado 1500 Extended Cab 2012
+74 Chevrolet Silverado 1500 Regular Cab 2012
+75 Chrysler Aspen SUV 2009
+76 Chrysler Sebring Convertible 2010
+78 Chrysler 300 SRT-8 2010
+82 Dodge Caliber Wagon 2012
+83 Dodge Caliber Wagon 2007
+84 Dodge Caravan Minivan 1997
+85 Dodge Ram Pickup 3500 Crew Cab 2010
+86 Dodge Ram Pickup 3500 Quad Cab 2009
+87 Dodge Sprinter Cargo Van 2009
+88 Dodge Journey SUV 2012
+89 Dodge Dakota Crew Cab 2010
+90 Dodge Dakota Club Cab 2007
+92 Dodge Challenger SRT8 2011
+93 Dodge Durango SUV 2012
+95 Dodge Charger Sedan 2012
+96 Dodge Charger SRT-8 2009
+98 FIAT 500 Abarth 2012
+99 FIAT 500 Convertible 2012
+100 Ferrari FF Coupe 2012
+101 Ferrari California Convertible 2012
+102 Ferrari 458 Italia Convertible 2012
+103 Ferrari 458 Italia Coupe 2012
+104 Fisker Karma Sedan 2012
+105 Ford F-450 Super Duty Crew Cab 2012
+106 Ford Mustang Convertible 2007
+107 Ford Freestar Minivan 2007
+108 Ford Expedition EL SUV 2009
+109 Ford Edge SUV 2012
+110 Ford Ranger SuperCab 2011
+111 Ford GT Coupe 2006
+112 Ford F-150 Regular Cab 2012
+113 Ford F-150 Regular Cab 2007
+114 Ford Focus Sedan 2007
+115 Ford E-Series Wagon Van 2012
+116 Ford Fiesta Sedan 2012
+118 GMC Savana Van 2012
+119 GMC Yukon Hybrid SUV 2012
+120 GMC Acadia SUV 2012
+121 GMC Canyon Extended Cab 2012
+122 Geo Metro Convertible 1993
+123 HUMMER H3T Crew Cab 2010
+124 HUMMER H2 SUT Crew Cab 2009
+125 Honda Odyssey Minivan 2012
+126 Honda Odyssey Minivan 2007
+127 Honda Accord Coupe 2012
+129 Hyundai Veloster Hatchback 2012
+130 Hyundai Santa Fe SUV 2012
+131 Hyundai Tucson SUV 2012
+132 Hyundai Veracruz SUV 2012
+133 Hyundai Sonata Hybrid Sedan 2012
+134 Hyundai Elantra Sedan 2007
+135 Hyundai Accent Sedan 2012
+136 Hyundai Genesis Sedan 2012
+137 Hyundai Sonata Sedan 2012
+138 Hyundai Elantra Touring Hatchback 2012
+139 Hyundai Azera Sedan 2012
+140 Infiniti G Coupe IPL 2012
+141 Infiniti QX56 SUV 2011
+142 Isuzu Ascender SUV 2008
+143 Jaguar XK XKR 2012
+144 Jeep Patriot SUV 2012
+145 Jeep Wrangler SUV 2012
+146 Jeep Liberty SUV 2012
+147 Jeep Grand Cherokee SUV 2012
+148 Jeep Compass SUV 2012
+150 Lamborghini Aventador Coupe 2012
+152 Lamborghini Diablo Coupe 2001
+153 Land Rover Range Rover SUV 2012
+154 Land Rover LR2 SUV 2012
+155 Lincoln Town Car Sedan 2011
+157 Maybach Landaulet Convertible 2012
+159 McLaren MP4-12C Coupe 2012
+160 Mercedes-Benz 300-Class Convertible 1993
+161 Mercedes-Benz C-Class Sedan 2012
+162 Mercedes-Benz SL-Class Coupe 2009
+164 Mercedes-Benz S-Class Sedan 2012
+165 Mercedes-Benz Sprinter Van 2012
+166 Mitsubishi Lancer Sedan 2012
+167 Nissan Leaf Hatchback 2012
+168 Nissan NV Passenger Van 2012
+169 Nissan Juke Hatchback 2012
+170 Nissan 240SX Coupe 1998
+171 Plymouth Neon Coupe 1999
+172 Porsche Panamera Sedan 2012
+173 Ram C/V Cargo Van Minivan 2012
+174 Rolls-Royce Phantom Drophead Coupe Convertible 2012
+175 Rolls-Royce Ghost Sedan 2012
+176 Rolls-Royce Phantom Sedan 2012
+177 Scion xD Hatchback 2012
+178 Spyker C8 Convertible 2009
+179 Spyker C8 Coupe 2009
+180 Suzuki Aerio Sedan 2007
+181 Suzuki Kizashi Sedan 2012
+182 Suzuki SX4 Hatchback 2012
+183 Suzuki SX4 Sedan 2012
+184 Tesla Model S Sedan 2012
+185 Toyota Sequoia SUV 2012
+186 Toyota Camry Sedan 2012
+187 Toyota Corolla Sedan 2012
+188 Toyota 4Runner SUV 2012
+189 Volkswagen Golf Hatchback 2012
+190 Volkswagen Golf Hatchback 1991
+191 Volkswagen Beetle Hatchback 2012
+192 Volvo C30 Hatchback 2012
+193 Volvo 240 Sedan 1993
+194 Volvo XC90 SUV 2007
+195 smart fortwo Convertible 2012

app.py

@@ -1,7 +1,270 @@
 import gradio as gr
+from PIL import Image
+import torch
+from torchvision import transforms
 
-def greet(name):
-    return "Hello " + name + "!!"
+# --------------------------
+# 1. Define your model class
+# --------------------------
+# Make sure this matches the architecture you used to train your model
+class MyCarClassifier(torch.nn.Module):
+    def __init__(self, num_classes=196):  # Stanford Cars has 196 classes
+        super(MyCarClassifier, self).__init__()
+        # Example: replace with your actual model architecture
+        from torchvision.models import resnet18
+        self.model = resnet18(pretrained=False)
+        self.model.fc = torch.nn.Linear(self.model.fc.in_features, num_classes)
+    
+    def forward(self, x):
+        return self.model(x)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# --------------------------
+# 2. Load model weights
+# --------------------------
+model = MyCarClassifier()
+model.load_state_dict(torch.load("best_stanford_cars_transfer_model.pth", map_location="cpu"))
+model.eval()  # important for inference
+
+# --------------------------
+# 3. Define labels
+# --------------------------
+# Full integer-to-label mapping for Stanford Cars dataset
+labels = [
+    "AM General Hummer SUV 2000",
+    "Acura RL Sedan 2012",
+    "Acura TL Sedan 2012",
+    "Acura TL Type-S 2008",
+    "Acura TSX Sedan 2012",
+    "Acura Integra Type R 2001",
+    "Acura ZDX Hatchback 2012",
+    "Aston Martin V8 Vantage Convertible 2012",
+    "Aston Martin V8 Vantage Coupe 2012",
+    "Aston Martin Virage Convertible 2012",
+    "Aston Martin Virage Coupe 2012",
+    "Audi RS 4 Convertible 2008",
+    "Audi A5 Coupe 2012",
+    "Audi TTS Coupe 2012",
+    "Audi R8 Coupe 2012",
+    "Audi V8 Sedan 1994",
+    "Audi 100 Sedan 1994",
+    "Audi 100 Wagon 1994",
+    "Audi TT Hatchback 2011",
+    "Audi S6 Sedan 2011",
+    "Audi S5 Convertible 2012",
+    "Audi S5 Coupe 2012",
+    "Audi S4 Sedan 2012",
+    "Audi S4 Sedan 2007",
+    "Audi TT RS Coupe 2012",
+    "BMW ActiveHybrid 5 Sedan 2012",
+    "BMW 1 Series Convertible 2012",
+    "BMW 1 Series Coupe 2012",
+    "BMW 3 Series Sedan 2012",
+    "BMW 3 Series Wagon 2012",
+    "BMW 6 Series Convertible 2007",
+    "BMW X5 SUV 2007",
+    "BMW X6 SUV 2012",
+    None,
+    None,
+    "BMW M6 Convertible 2010",
+    "BMW X3 SUV 2012",
+    "BMW Z4 Convertible 2012",
+    "Bentley Continental Supersports Conv. Convertible 2012",
+    "Bentley Arnage Sedan 2009",
+    "Bentley Mulsanne Sedan 2011",
+    "Bentley Continental GT Coupe 2012",
+    "Bentley Continental GT Coupe 2007",
+    "Bentley Continental Flying Spur Sedan 2007",
+    "Bugatti Veyron 16.4 Convertible 2009",
+    "Bugatti Veyron 16.4 Coupe 2009",
+    "Buick Regal GS 2012",
+    "Buick Rainier SUV 2007",
+    "Buick Verano Sedan 2012",
+    "Buick Enclave SUV 2012",
+    "Cadillac CTS-V Sedan 2012",
+    None,
+    "Cadillac Escalade EXT Crew Cab 2007",
+    "Chevrolet Silverado 1500 Hybrid Crew Cab 2012",
+    "Chevrolet Corvette Convertible 2012",
+    "Chevrolet Corvette ZR1 2012",
+    "Chevrolet Corvette Ron Fellows Edition Z06 2007",
+    "Chevrolet Traverse SUV 2012",
+    "Chevrolet Camaro Convertible 2012",
+    "Chevrolet HHR SS 2010",
+    "Chevrolet Impala Sedan 2007",
+    "Chevrolet Tahoe Hybrid SUV 2012",
+    "Chevrolet Sonic Sedan 2012",
+    "Chevrolet Express Cargo Van 2007",
+    "Chevrolet Avalanche Crew Cab 2012",
+    "Chevrolet Cobalt SS 2010",
+    "Chevrolet Malibu Hybrid Sedan 2010",
+    "Chevrolet TrailBlazer SS 2009",
+    "Chevrolet Silverado 2500HD Regular Cab 2012",
+    "Chevrolet Silverado 1500 Classic Extended Cab 2007",
+    "Chevrolet Express Van 2007",
+    "Chevrolet Monte Carlo Coupe 2007",
+    "Chevrolet Malibu Sedan 2007",
+    "Chevrolet Silverado 1500 Extended Cab 2012",
+    "Chevrolet Silverado 1500 Regular Cab 2012",
+    "Chrysler Aspen SUV 2009",
+    "Chrysler Sebring Convertible 2010",
+    None,
+    "Chrysler 300 SRT-8 2010",
+    None,
+    None,
+    "Dodge Caliber Wagon 2012",
+    "Dodge Caliber Wagon 2007",
+    "Dodge Caravan Minivan 1997",
+    "Dodge Ram Pickup 3500 Crew Cab 2010",
+    "Dodge Ram Pickup 3500 Quad Cab 2009",
+    "Dodge Sprinter Cargo Van 2009",
+    "Dodge Journey SUV 2012",
+    "Dodge Dakota Crew Cab 2010",
+    "Dodge Dakota Club Cab 2007",
+    None,
+    "Dodge Challenger SRT8 2011",
+    "Dodge Durango SUV 2012",
+    None,
+    "Dodge Charger Sedan 2012",
+    "Dodge Charger SRT-8 2009",
+    None,
+    "FIAT 500 Abarth 2012",
+    "FIAT 500 Convertible 2012",
+    "Ferrari FF Coupe 2012",
+    "Ferrari California Convertible 2012",
+    "Ferrari 458 Italia Convertible 2012",
+    "Ferrari 458 Italia Coupe 2012",
+    "Fisker Karma Sedan 2012",
+    "Ford F-450 Super Duty Crew Cab 2012",
+    "Ford Mustang Convertible 2007",
+    "Ford Freestar Minivan 2007",
+    "Ford Expedition EL SUV 2009",
+    "Ford Edge SUV 2012",
+    "Ford Ranger SuperCab 2011",
+    "Ford GT Coupe 2006",
+    "Ford F-150 Regular Cab 2012",
+    "Ford F-150 Regular Cab 2007",
+    "Ford Focus Sedan 2007",
+    "Ford E-Series Wagon Van 2012",
+    "Ford Fiesta Sedan 2012",
+    None,
+    "GMC Savana Van 2012",
+    "GMC Yukon Hybrid SUV 2012",
+    "GMC Acadia SUV 2012",
+    "GMC Canyon Extended Cab 2012",
+    "Geo Metro Convertible 1993",
+    "HUMMER H3T Crew Cab 2010",
+    "HUMMER H2 SUT Crew Cab 2009",
+    "Honda Odyssey Minivan 2012",
+    "Honda Odyssey Minivan 2007",
+    "Honda Accord Coupe 2012",
+    None,
+    "Hyundai Veloster Hatchback 2012",
+    "Hyundai Santa Fe SUV 2012",
+    "Hyundai Tucson SUV 2012",
+    "Hyundai Veracruz SUV 2012",
+    "Hyundai Sonata Hybrid Sedan 2012",
+    "Hyundai Elantra Sedan 2007",
+    "Hyundai Accent Sedan 2012",
+    "Hyundai Genesis Sedan 2012",
+    "Hyundai Sonata Sedan 2012",
+    "Hyundai Elantra Touring Hatchback 2012",
+    "Hyundai Azera Sedan 2012",
+    "Infiniti G Coupe IPL 2012",
+    "Infiniti QX56 SUV 2011",
+    "Isuzu Ascender SUV 2008",
+    "Jaguar XK XKR 2012",
+    "Jeep Patriot SUV 2012",
+    "Jeep Wrangler SUV 2012",
+    "Jeep Liberty SUV 2012",
+    "Jeep Grand Cherokee SUV 2012",
+    "Jeep Compass SUV 2012",
+    None,
+    "Lamborghini Aventador Coupe 2012",
+    None,
+    "Lamborghini Diablo Coupe 2001",
+    "Land Rover Range Rover SUV 2012",
+    "Land Rover LR2 SUV 2012",
+    "Lincoln Town Car Sedan 2011",
+    None,
+    "Maybach Landaulet Convertible 2012",
+    None,
+    "McLaren MP4-12C Coupe 2012",
+    "Mercedes-Benz 300-Class Convertible 1993",
+    "Mercedes-Benz C-Class Sedan 2012",
+    "Mercedes-Benz SL-Class Coupe 2009",
+    None,
+    "Mercedes-Benz S-Class Sedan 2012",
+    "Mercedes-Benz Sprinter Van 2012",
+    "Mitsubishi Lancer Sedan 2012",
+    "Nissan Leaf Hatchback 2012",
+    "Nissan NV Passenger Van 2012",
+    "Nissan Juke Hatchback 2012",
+    "Nissan 240SX Coupe 1998",
+    "Plymouth Neon Coupe 1999",
+    "Porsche Panamera Sedan 2012",
+    "Ram C/V Cargo Van Minivan 2012",
+    "Rolls-Royce Phantom Drophead Coupe Convertible 2012",
+    "Rolls-Royce Ghost Sedan 2012",
+    "Rolls-Royce Phantom Sedan 2012",
+    "Scion xD Hatchback 2012",
+    "Spyker C8 Convertible 2009",
+    "Spyker C8 Coupe 2009",
+    "Suzuki Aerio Sedan 2007",
+    "Suzuki Kizashi Sedan 2012",
+    "Suzuki SX4 Hatchback 2012",
+    "Suzuki SX4 Sedan 2012",
+    "Tesla Model S Sedan 2012",
+    "Toyota Sequoia SUV 2012",
+    "Toyota Camry Sedan 2012",
+    "Toyota Corolla Sedan 2012",
+    "Toyota 4Runner SUV 2012",
+    "Volkswagen Golf Hatchback 2012",
+    "Volkswagen Golf Hatchback 1991",
+    "Volkswagen Beetle Hatchback 2012",
+    "Volvo C30 Hatchback 2012",
+    "Volvo 240 Sedan 1993",
+    "Volvo XC90 SUV 2007",
+    "smart fortwo Convertible 2012"
+]
+
+# --------------------------
+# 4. Preprocessing function
+# --------------------------
+def preprocess_image(img: Image.Image):
+    transform = transforms.Compose([
+        transforms.Resize((224, 224)),           # match your training input size
+        transforms.ToTensor(),
+        transforms.Normalize(
+            mean=[0.485, 0.456, 0.406],         # if you used ImageNet pretrained normalization
+            std=[0.229, 0.224, 0.225]
+        )
+    ])
+    x = transform(img).unsqueeze(0)  # add batch dimension
+    return x
+
+# --------------------------
+# 5. Prediction function
+# --------------------------
+def predict(img: Image.Image):
+    x = preprocess_image(img)
+    with torch.no_grad():
+        logits = model(x)
+        probs = torch.nn.functional.softmax(logits, dim=1)[0]
+        top_idx = torch.argmax(probs).item()
+        top_label = labels[top_idx] if labels[top_idx] is not None else f"Class {top_idx}"
+        confidence = float(probs[top_idx])
+        return {top_label: confidence}
+
+# --------------------------
+# 6. Gradio interface
+# --------------------------
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Label(num_top_classes=1),
+    title="Stanford Cars Classifier",
+    description="Upload a car image and see the most likely predicted make/model and its confidence."
+)
+
+if __name__ == "__main__":
+    demo.launch()