Deploy the App

app.py

@@ -0,0 +1,34 @@
+import gradio as gr
+import torch
+from torchvision import transforms
+
+model = torch.jit.load("../models/mobilenet.pt")
+model.eval()
+
+transform = transforms.Compose([
+  transforms.RandomResizedCrop(224),
+  transforms.RandomHorizontalFlip(),
+  transforms.ToTensor(),
+  transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225])
+])
+
+CLASSES = ["Ak", "Ala_Idris", "Buzgulu", "Dimnit", "Nazli"]
+
+def classify_image(inp):
+  inp = transform(inp).unsqueeze(0)
+  out = model(inp)
+  # print(out.argmax(dim=1, keepdim=True))
+  # print(out.argmax())
+  return CLASSES[out.argmax().item()]
+
+iface = gr.Interface(fn=classify_image,
+                     inputs=gr.Image(type="pil", label="Input Image"),
+                     outputs="text",
+                     examples=[
+                       "../data/app_data/ak.png",
+                       "../data/app_data/idris.png",
+                       "../data/app_data/buzgulu.png",
+                       "../data/app_data/dimnit.png",
+                       "../data/app_data/nazli.png",
+                     ])
+iface.launch()

models/mobilenet.pt

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