Updated to run on cpu.

README.md

@@ -1,13 +1,13 @@
 ---
 title: Stanford Cars
-emoji: 📈
-colorFrom: pink
-colorTo: red
+emoji: 👀
+colorFrom: red
+colorTo: yellow
 sdk: gradio
-sdk_version: 4.21.0
+sdk_version: 3.41.0
 app_file: app.py
 pinned: false
-license: apache-2.0
+license: mit
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,61 @@
+import gradio as gr
+
+from pathlib import Path
+import openvino as ov
+import mmcv
+import numpy as np
+
+with open("/model_config/car_classes.txt", "r") as f:
+    car_classes = f.readlines()
+CAR_CLASSES = [c.replace(",\n", "").replace("'","") for c in car_classes]
+
+def pred_class(image_path):
+
+    base_artifacts_dir = Path("./")
+
+    model_name = "end2end"
+    model_xml_name = f'{model_name}.xml'
+    model_bin_name = f'{model_name}.bin'
+
+    model_xml_path = base_artifacts_dir / model_xml_name
+    device="CPU"
+    
+    core = ov.Core()
+    model = core.read_model(model=model_xml_path)
+    compiled_model = core.compile_model(model=model, device_name=device)
+
+    output_layer = compiled_model.output(0)
+
+    image = mmcv.imread(image_path)
+
+    # Resize to MobileNet image shape.
+    input_image = mmcv.imresize(image, (224, 224))
+    normalized_image = mmcv.imnormalize(input_image, mean=np.array([123.675, 116.28, 103.53]), std=np.array([58.395, 57.12, 57.375]))
+
+    # Reshape to model input shape.
+    input_image = np.expand_dims(np.transpose(normalized_image, (2,0,1)), 0)
+    result_infer = compiled_model([input_image])[output_layer]
+    result_index = np.argmax(result_infer)
+    
+    return CAR_CLASSES[result_index]
+
+inputs_image = [
+    gr.components.Image(type="filepath", label="Input Image"),
+]
+outputs_text = [
+    gr.components.Textbox(label="Output Text"),
+]
+interface_image = gr.Interface(
+    fn=pred_class,
+    inputs=inputs_image,
+    outputs=outputs_text,
+    title="Car Classification",
+    cache_examples=False,
+)
+def greet(name):
+    return "Hello " + name + "!!"
+
+gr.TabbedInterface(
+    [interface_image],
+    tab_names=['Image inference']
+).queue().launch()

car_classes.txt

@@ -0,0 +1,196 @@
+'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',
+'BMW M3 Coupe 2012',
+'BMW M5 Sedan 2010',
+'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',
+'Cadillac SRX SUV 2012',
+'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',
+'Chrysler Town and Country Minivan 2012',
+'Chrysler 300 SRT-8 2010',
+'Chrysler Crossfire Convertible 2008',
+'Chrysler PT Cruiser Convertible 2008',
+'Daewoo Nubira Wagon 2002',
+'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',
+'Dodge Magnum Wagon 2008',
+'Dodge Challenger SRT8 2011',
+'Dodge Durango SUV 2012',
+'Dodge Durango SUV 2007',
+'Dodge Charger Sedan 2012',
+'Dodge Charger SRT-8 2009',
+'Eagle Talon Hatchback 1998',
+'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',
+'GMC Terrain SUV 2012',
+'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',
+'Honda Accord Sedan 2012',
+'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',
+'Lamborghini Reventon Coupe 2008',
+'Lamborghini Aventador Coupe 2012',
+'Lamborghini Gallardo LP 570-4 Superleggera 2012',
+'Lamborghini Diablo Coupe 2001',
+'Land Rover Range Rover SUV 2012',
+'Land Rover LR2 SUV 2012',
+'Lincoln Town Car Sedan 2011',
+'MINI Cooper Roadster Convertible 2012',
+'Maybach Landaulet Convertible 2012',
+'Mazda Tribute SUV 2011',
+'McLaren MP4-12C Coupe 2012',
+'Mercedes-Benz 300-Class Convertible 1993',
+'Mercedes-Benz C-Class Sedan 2012',
+'Mercedes-Benz SL-Class Coupe 2009',
+'Mercedes-Benz E-Class Sedan 2012',
+'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'

end2end.bin

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

requirements.txt

@@ -0,0 +1,3 @@
+mmcv
+openvino
+