init commit

app.py

@@ -0,0 +1,18 @@
+from fastai.vision.all import *
+import gradio as gr
+
+categories = ['Abstract', 'African', 'American Colonial', 'Amish', 'Arabian', 'Art Deco', 'Art Moderne', 'Art Nouveau', 'Artisan', 'Arts & Crafts', 'Asian', 'Baroque', 'Bauhaus', 'Beach House', 'Bohemian', 'Brazilian', 'British Colonial', 'Carolean', 'Chinese', 'Chippendale', 'Coastal', 'Commonwealth', 'Contemporary', 'Cool Britannia', 'Cottage', 'Country', 'Danish', 'Directoire', 'Dutch Renaissance', 'Eclectic', 'Egyptian', 'Elizabethan', 'Empire', 'English', 'English Country', 'European', 'Exploration', 'Farmhouse', 'Finnish', 'Flemish', 'French', 'French Provincial', 'Georgian', 'Gothic', 'Greek', 'Hygge', 'Indian', 'Industrial', 'Italian', 'Jacobean', 'Japandi', 'Japanese', 'Lake House', 'Machine Age', 'Maximalist', 'Medieval', 'Mediterranean', 'Memphis', 'Mexican', 'Mid-Century Modern', 'Minimalist', 'Mission', 'Modern', 'Modernist', 'Moroccan', 'Nautical', 'Neoclassic', 'Northwestern', 'Old World', 'Organic', 'Palladian', 'Parisian', 'Pennsylvania Dutch', 'Plantation', 'Post-Modern', 'Puritan', 'Queen Anne', 'Regal', 'Regence', 'Regency', 'Renaissance', 'Retro', 'Revival', 'Rietveld', 'Rococo', 'Romantic', 'Russian', 'Rustic', 'Scandinavian', 'Shabby Chic', 'Shaker', 'Southwestern', 'Space Age', 'Spanish Renaissance', 'Steampunk', 'Swedish', 'Traditional', 'Transitional', 'Tropical', 'Tudor', 'Tuscan', 'Urban', 'Venetian', 'Victorian', 'Vintage', 'Western', 'William & Mary', 'Zen']
+
+learn = load_learner('model.pkl')
+
+def classify_image(img):
+    pred, idx, probs = learn.predict(img)
+    return dict(zip(categories, map(float,probs)))
+
+
+image = gr.inputs.Image()
+label = gr.outputs.Label()
+examples = ['Japandi.jpg', 'Amish.jpg']
+
+intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
+intf.launch(inline=False)