Update app.py

app.py

@@ -12,12 +12,12 @@ model.load_state_dict(state_dict)
 model.eval()
 
 #predictions
-imgTransforms = transforms.Compose([transforms.Resize(256), transforms.CenterCrop(224), trasforms.ToTensor(),transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225])])
+imgTransforms = transforms.Compose([transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor(),transforms.Normalize([0.485,0.456,0.406],[0.229,0.224,0.225])])
 
 LABELS =['Fiat 500', 'VW Up']
 def predict(inp):
     inp= Image.fromarray(inp.astype('unit8'), 'RGB')
-    inp =imgTransfroms(inp).unsqueeze(0)
+    inp =imgTransforms(inp).unsqueeze(0)
 
     with torch.no_grad():
         predictions = torch.nn.functional.softmax(model(inp)[0])