initial commit

app.py

@@ -0,0 +1,40 @@
+import torch
+import torch.nn.functional as F
+from torchvision import transforms
+from PIL import Image
+import timm
+import json
+import gradio as gr
+
+DEVICE = "cpu"
+IMG_SIZE = 384
+
+# Load classes
+with open('classes.json', 'r') as f:
+    classes = json.load(f)
+
+def load_model():
+    model = timm.create_model("tf_efficientnetv2_s", pretrained=False, num_classes=31, drop_rate=0.3)
+    ckpt = torch.load('model.pt', map_location=DEVICE)
+    model.load_state_dict(ckpt['model_state_dict'])
+    model.eval()
+    return model
+
+model = load_model()
+
+val_tfms = transforms.Compose([
+    transforms.Resize((IMG_SIZE, IMG_SIZE)),
+    transforms.ToTensor(),
+])
+
+def predict(image):
+    img = val_tfms(image).unsqueeze(0).to(DEVICE)
+    with torch.no_grad():
+        logits = model(img)
+        probs = F.softmax(logits, dim=1)
+        top_prob, top_idx = probs.topk(1, dim=1)
+        idx_to_class = {v:k for k,v in classes.items()}
+        label = idx_to_class[top_idx[0][0].item()]
+        return f"{label} ({top_prob[0][0].item():.2f})"
+
+gr.Interface(fn=predict, inputs=gr.Image(type="pil"), outputs="text").launch()

classes.json

@@ -0,0 +1,33 @@
+{
+  "Appam": 0,
+  "Beetroot poriyal": 1,
+  "Boiled Egg": 2,
+  "Carrot poriyal": 3,
+  "Chicken 65": 4,
+  "Chicken briyani": 5,
+  "Dosa": 6,
+  "Idly": 7,
+  "Kaara chutney": 8,
+  "Kali": 9,
+  "Koozh": 10,
+  "Lemon Rice": 11,
+  "Mushroom briyani": 12,
+  "Mutton Briyani": 13,
+  "Nandu masala": 14,
+  "Nei satham": 15,
+  "Paal kolukattai": 16,
+  "Paneer briyani": 17,
+  "Panner masala": 18,
+  "Parupu vada": 19,
+  "Pidi kolukattai": 20,
+  "Poorna kolukattai": 21,
+  "Prawn thokku": 22,
+  "Puthina Chutney": 23,
+  "Sambar": 24,
+  "Sambar satham": 25,
+  "Satham": 26,
+  "Thengai chutney": 27,
+  "Uzhuntha vadai": 28,
+  "Veg briyani": 29,
+  "Ven Pongal": 30
+}

model.pt

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

requirements.txt

@@ -0,0 +1,5 @@
+torch>=2.1.0
+torchvision>=0.18.0
+timm>=0.9.0
+Pillow>=10.0.0
+gradio>=3.40.0