Add application file

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+import torch
+from model import model, classes
+
+# Load state dict
+model.load_state_dict(torch.load('model.pth'))
+
+# Transforms
+transform = transforms.Compose([
+    transforms.Resize((32, 32)),
+    transforms.ToTensor(),
+    transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
+
+def classify(img):
+    x = transform(img)
+
+    with torch.no_grad():
+        pred = model(x)
+
+    idx = torch.argmax(pred)
+    cls = classes[idx]
+
+    return cls
+
+
+iface = gr.Interface(...)
+iface.launch()