Initial commit

.gitignore

@@ -0,0 +1 @@
+.DS_Store

app.py

@@ -0,0 +1,81 @@
+import pickle
+import random
+import pandas as pd
+import gradio as gr
+from fastai.vision.all import *
+
+zone_lookup = pd.read_csv('./data/zone_lookup.csv')
+
+with open('./models/lin_reg.bin', 'rb') as handle:
+    dv, model = pickle.load(handle)
+
+def prepare_features(pickup, dropoff, trip_distance):
+    pickupId = zone_lookup[zone_lookup["borough_zone"] == pickup].LocationID
+    dropoffId = zone_lookup[zone_lookup["borough_zone"] == dropoff].LocationID
+    trip_distance = round(trip_distance, 4)
+
+    features = {}
+    features['PU_DO'] = '%s_%s' % (pickupId, dropoffId)
+    features['trip_distance'] = trip_distance
+    return features
+
+def predict(pickup, dropoff, trip_distance):
+    features = prepare_features(pickup, dropoff, trip_distance)
+    X = dv.transform(features)
+    preds = model.predict(X)
+    duration = float(preds[0])
+    return "The predicted duration is %.4f minutes." % duration
+
+with gr.Blocks() as demo:
+    gr.Markdown("Predict Taxi Duration or Classify dog breeds using this demo")
+
+    with gr.Tab("Predict Taxi Duration"):
+        with gr.Row():
+            pickup = gr.Dropdown(
+                choices=list(zone_lookup["borough_zone"]), 
+                label="Pickup Location", 
+                info="The location where the passenger(s) were picked up", 
+                value=lambda: random.choice(zone_lookup["borough_zone"])
+            )
+
+            dropoff = gr.Dropdown(
+                choices=list(zone_lookup["borough_zone"]), 
+                label="Dropoff Location", 
+                info="The location where the passenger(s) were dropped off", 
+                value=lambda: random.choice(zone_lookup["borough_zone"])
+            )
+
+        trip_distance = gr.Slider(
+            minimum=0.0, 
+            maximum=100.0,
+            step=0.1,
+            label="Trip Distance",
+            info="The trip distance in miles calculated by the taximeter",
+            value=lambda: random.uniform(0.0, 100.0)
+        )
+        with gr.Column():
+            output = gr.Textbox(label="Output Box")
+            predict_btn = gr.Button("Predict")
+    
+    with gr.Tab("Classify Dog Breed"):
+        def is_cat(x): return x[0].isupper()
+
+        learn = load_learner('./models/model.pkl')
+
+        categories = ('Dog', 'Cat')
+
+        def classify_image(img):
+            pred, idx, probs = learn.predict(img)
+            return dict(zip(categories, map(float,probs)))
+
+        image = gr.inputs.Image(shape=(192, 192))
+        label = gr.outputs.Label()
+        examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
+        
+        classify_btn = gr.Button("Predict")
+
+
+    predict_btn.click(fn=predict, inputs=[pickup, dropoff, trip_distance], outputs=output, api_name="predict-duration")       
+    classify_btn.click(fn=classify_image, inputs=image, outputs=label, api_name="classify-dog-breed")
+
+demo.launch()

data/zone_lookup.csv

@@ -0,0 +1,261 @@
+LocationID,borough_zone
+1,EWR - Newark Airport
+2,Queens - Jamaica Bay
+3,Bronx - Allerton/Pelham Gardens
+4,Manhattan - Alphabet City
+5,Staten Island - Arden Heights
+6,Staten Island - Arrochar/Fort Wadsworth
+7,Queens - Astoria
+8,Queens - Astoria Park
+9,Queens - Auburndale
+10,Queens - Baisley Park
+11,Brooklyn - Bath Beach
+12,Manhattan - Battery Park
+13,Manhattan - Battery Park City
+14,Brooklyn - Bay Ridge
+15,Queens - Bay Terrace/Fort Totten
+16,Queens - Bayside
+17,Brooklyn - Bedford
+18,Bronx - Bedford Park
+19,Queens - Bellerose
+20,Bronx - Belmont
+21,Brooklyn - Bensonhurst East
+22,Brooklyn - Bensonhurst West
+23,Staten Island - Bloomfield/Emerson Hill
+24,Manhattan - Bloomingdale
+25,Brooklyn - Boerum Hill
+26,Brooklyn - Borough Park
+27,Queens - Breezy Point/Fort Tilden/Riis Beach
+28,Queens - Briarwood/Jamaica Hills
+29,Brooklyn - Brighton Beach
+30,Queens - Broad Channel
+31,Bronx - Bronx Park
+32,Bronx - Bronxdale
+33,Brooklyn - Brooklyn Heights
+34,Brooklyn - Brooklyn Navy Yard
+35,Brooklyn - Brownsville
+36,Brooklyn - Bushwick North
+37,Brooklyn - Bushwick South
+38,Queens - Cambria Heights
+39,Brooklyn - Canarsie
+40,Brooklyn - Carroll Gardens
+41,Manhattan - Central Harlem
+42,Manhattan - Central Harlem North
+43,Manhattan - Central Park
+44,Staten Island - Charleston/Tottenville
+45,Manhattan - Chinatown
+46,Bronx - City Island
+47,Bronx - Claremont/Bathgate
+48,Manhattan - Clinton East
+49,Brooklyn - Clinton Hill
+50,Manhattan - Clinton West
+51,Bronx - Co-Op City
+52,Brooklyn - Cobble Hill
+53,Queens - College Point
+54,Brooklyn - Columbia Street
+55,Brooklyn - Coney Island
+56,Queens - Corona
+58,Bronx - Country Club
+59,Bronx - Crotona Park
+60,Bronx - Crotona Park East
+61,Brooklyn - Crown Heights North
+62,Brooklyn - Crown Heights South
+63,Brooklyn - Cypress Hills
+64,Queens - Douglaston
+65,Brooklyn - Downtown Brooklyn/MetroTech
+66,Brooklyn - DUMBO/Vinegar Hill
+67,Brooklyn - Dyker Heights
+68,Manhattan - East Chelsea
+69,Bronx - East Concourse/Concourse Village
+70,Queens - East Elmhurst
+71,Brooklyn - East Flatbush/Farragut
+72,Brooklyn - East Flatbush/Remsen Village
+73,Queens - East Flushing
+74,Manhattan - East Harlem North
+75,Manhattan - East Harlem South
+76,Brooklyn - East New York
+77,Brooklyn - East New York/Pennsylvania Avenue
+78,Bronx - East Tremont
+79,Manhattan - East Village
+80,Brooklyn - East Williamsburg
+81,Bronx - Eastchester
+82,Queens - Elmhurst
+83,Queens - Elmhurst/Maspeth
+84,Staten Island - Eltingville/Annadale/Prince's Bay
+85,Brooklyn - Erasmus
+86,Queens - Far Rockaway
+87,Manhattan - Financial District North
+88,Manhattan - Financial District South
+89,Brooklyn - Flatbush/Ditmas Park
+90,Manhattan - Flatiron
+91,Brooklyn - Flatlands
+92,Queens - Flushing
+93,Queens - Flushing Meadows-Corona Park
+94,Bronx - Fordham South
+95,Queens - Forest Hills
+96,Queens - Forest Park/Highland Park
+97,Brooklyn - Fort Greene
+98,Queens - Fresh Meadows
+99,Staten Island - Freshkills Park
+100,Manhattan - Garment District
+101,Queens - Glen Oaks
+102,Queens - Glendale
+103,Manhattan - Governor's Island/Ellis Island/Liberty Island
+106,Brooklyn - Gowanus
+107,Manhattan - Gramercy
+108,Brooklyn - Gravesend
+109,Staten Island - Great Kills
+110,Staten Island - Great Kills Park
+111,Brooklyn - Green-Wood Cemetery
+112,Brooklyn - Greenpoint
+113,Manhattan - Greenwich Village North
+114,Manhattan - Greenwich Village South
+115,Staten Island - Grymes Hill/Clifton
+116,Manhattan - Hamilton Heights
+117,Queens - Hammels/Arverne
+118,Staten Island - Heartland Village/Todt Hill
+119,Bronx - Highbridge
+120,Manhattan - Highbridge Park
+121,Queens - Hillcrest/Pomonok
+122,Queens - Hollis
+123,Brooklyn - Homecrest
+124,Queens - Howard Beach
+125,Manhattan - Hudson Sq
+126,Bronx - Hunts Point
+127,Manhattan - Inwood
+128,Manhattan - Inwood Hill Park
+129,Queens - Jackson Heights
+130,Queens - Jamaica
+131,Queens - Jamaica Estates
+132,Queens - JFK Airport
+133,Brooklyn - Kensington
+134,Queens - Kew Gardens
+135,Queens - Kew Gardens Hills
+136,Bronx - Kingsbridge Heights
+137,Manhattan - Kips Bay
+138,Queens - LaGuardia Airport
+139,Queens - Laurelton
+140,Manhattan - Lenox Hill East
+141,Manhattan - Lenox Hill West
+142,Manhattan - Lincoln Square East
+143,Manhattan - Lincoln Square West
+144,Manhattan - Little Italy/NoLiTa
+145,Queens - Long Island City/Hunters Point
+146,Queens - Long Island City/Queens Plaza
+147,Bronx - Longwood
+148,Manhattan - Lower East Side
+149,Brooklyn - Madison
+150,Brooklyn - Manhattan Beach
+151,Manhattan - Manhattan Valley
+152,Manhattan - Manhattanville
+153,Manhattan - Marble Hill
+154,Brooklyn - Marine Park/Floyd Bennett Field
+155,Brooklyn - Marine Park/Mill Basin
+156,Staten Island - Mariners Harbor
+157,Queens - Maspeth
+158,Manhattan - Meatpacking/West Village West
+159,Bronx - Melrose South
+160,Queens - Middle Village
+161,Manhattan - Midtown Center
+162,Manhattan - Midtown East
+163,Manhattan - Midtown North
+164,Manhattan - Midtown South
+165,Brooklyn - Midwood
+166,Manhattan - Morningside Heights
+167,Bronx - Morrisania/Melrose
+168,Bronx - Mott Haven/Port Morris
+169,Bronx - Mount Hope
+170,Manhattan - Murray Hill
+171,Queens - Murray Hill-Queens
+172,Staten Island - New Dorp/Midland Beach
+173,Queens - North Corona
+174,Bronx - Norwood
+175,Queens - Oakland Gardens
+176,Staten Island - Oakwood
+177,Brooklyn - Ocean Hill
+178,Brooklyn - Ocean Parkway South
+179,Queens - Old Astoria
+180,Queens - Ozone Park
+181,Brooklyn - Park Slope
+182,Bronx - Parkchester
+183,Bronx - Pelham Bay
+184,Bronx - Pelham Bay Park
+185,Bronx - Pelham Parkway
+186,Manhattan - Penn Station/Madison Sq West
+187,Staten Island - Port Richmond
+188,Brooklyn - Prospect-Lefferts Gardens
+189,Brooklyn - Prospect Heights
+190,Brooklyn - Prospect Park
+191,Queens - Queens Village
+192,Queens - Queensboro Hill
+193,Queens - Queensbridge/Ravenswood
+194,Manhattan - Randalls Island
+195,Brooklyn - Red Hook
+196,Queens - Rego Park
+197,Queens - Richmond Hill
+198,Queens - Ridgewood
+199,Bronx - Rikers Island
+200,Bronx - Riverdale/North Riverdale/Fieldston
+201,Queens - Rockaway Park
+202,Manhattan - Roosevelt Island
+203,Queens - Rosedale
+204,Staten Island - Rossville/Woodrow
+205,Queens - Saint Albans
+206,Staten Island - Saint George/New Brighton
+207,Queens - Saint Michaels Cemetery/Woodside
+208,Bronx - Schuylerville/Edgewater Park
+209,Manhattan - Seaport
+210,Brooklyn - Sheepshead Bay
+211,Manhattan - SoHo
+212,Bronx - Soundview/Bruckner
+213,Bronx - Soundview/Castle Hill
+214,Staten Island - South Beach/Dongan Hills
+215,Queens - South Jamaica
+216,Queens - South Ozone Park
+217,Brooklyn - South Williamsburg
+218,Queens - Springfield Gardens North
+219,Queens - Springfield Gardens South
+220,Bronx - Spuyten Duyvil/Kingsbridge
+221,Staten Island - Stapleton
+222,Brooklyn - Starrett City
+223,Queens - Steinway
+224,Manhattan - Stuy Town/Peter Cooper Village
+225,Brooklyn - Stuyvesant Heights
+226,Queens - Sunnyside
+227,Brooklyn - Sunset Park East
+228,Brooklyn - Sunset Park West
+229,Manhattan - Sutton Place/Turtle Bay North
+230,Manhattan - Times Sq/Theatre District
+231,Manhattan - TriBeCa/Civic Center
+232,Manhattan - Two Bridges/Seward Park
+233,Manhattan - UN/Turtle Bay South
+234,Manhattan - Union Sq
+235,Bronx - University Heights/Morris Heights
+236,Manhattan - Upper East Side North
+237,Manhattan - Upper East Side South
+238,Manhattan - Upper West Side North
+239,Manhattan - Upper West Side South
+240,Bronx - Van Cortlandt Park
+241,Bronx - Van Cortlandt Village
+242,Bronx - Van Nest/Morris Park
+243,Manhattan - Washington Heights North
+244,Manhattan - Washington Heights South
+245,Staten Island - West Brighton
+246,Manhattan - West Chelsea/Hudson Yards
+247,Bronx - West Concourse
+248,Bronx - West Farms/Bronx River
+249,Manhattan - West Village
+250,Bronx - Westchester Village/Unionport
+251,Staten Island - Westerleigh
+252,Queens - Whitestone
+253,Queens - Willets Point
+254,Bronx - Williamsbridge/Olinville
+255,Brooklyn - Williamsburg (North Side)
+256,Brooklyn - Williamsburg (South Side)
+257,Brooklyn - Windsor Terrace
+258,Queens - Woodhaven
+259,Bronx - Woodlawn/Wakefield
+260,Queens - Woodside
+261,Manhattan - World Trade Center
+262,Manhattan - Yorkville East
+263,Manhattan - Yorkville West

models/lin_reg.bin

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

models/model.pkl

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

requirements.txt

@@ -0,0 +1,3 @@
+fastai
+pandas
+scikit-learn