initial commit

app.py

@@ -0,0 +1,38 @@
+from typing import List, Tuple, Dict
+import torch
+import os
+from timeit import default_timer as timer
+import PIL
+import gradio as gr
+from model import create_effnetb2_model
+
+class_names = ['pizza', 'steak', 'sushi']
+examples = [os.path.join('examples', img) for img in os.listdir('examples')]
+
+model, preprocess = create_effnetb2_model(num_classes=3, seed=42)
+model.load_state_dict(torch.load('effnetb2_20_percent.pth'))
+
+def predict(img: PIL.Image) -> Tuple[Dict, float]:
+
+    start_time = timer()
+
+    img = preprocess(img).unsqueeze(dim=0)
+
+    model.to('cpu')
+    model.eval()
+    with torch.inference_mode():
+        probs = model(img).softmax(dim=-1).squeeze().tolist()
+        preds = {class_name: prob for class_name, prob in zip(class_names, probs)}
+    
+    pred_time = round(timer() - start_time, 8)
+
+    return preds, pred_time
+
+demo = gr.Interface(fn=predict,
+                    inputs=gr.Image(type='pil'),
+                    outputs=[gr.Label(num_top_classes=3, label='Prediction probabilities'),
+                             gr.Number(label='Prediction time (s)')],
+                    examples=examples,
+                    title='FoodVision Mini 🍕🥩🍣')
+
+demo.launch()

effnetb2_20_percent.pth

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

model.py

@@ -0,0 +1,20 @@
+
+import torch
+from torch import nn
+from torchvision.models import EfficientNet_B2_Weights, efficientnet_b2
+
+def create_effnetb2_model(num_classes: int = 3,
+                          seed: int = 42):
+    torch.manual_seed(seed)
+    torch.cuda.manual_seed(seed)
+
+    weights = EfficientNet_B2_Weights.DEFAULT
+    transforms = weights.transforms()
+    model = efficientnet_b2(weights=weights)
+
+    for param in model.parameters():
+        param.requires_grad = False
+    
+    model.classifier[1] = nn.Linear(in_features=1408, out_features=num_classes)
+
+    return model, transforms

requirements.txt

@@ -0,0 +1,3 @@
+torch==1.12.0
+torchvision==0.13.0
+gradio==3.1.4