Update app.py

app.py

@@ -55,7 +55,7 @@ def classifyCar(im):
   im = Image.fromarray(im.astype('uint8'), 'RGB')
   im = carTransforms(im).unsqueeze(0)  # transform and add batch dimension
   with torch.no_grad():
-    scores = torch.nn.functional.softmax(model(im)[0])
+    scores = torch.nn.functional.softmax(DesignModernityModel(im)[0])
   return {LABELS[i]: float(scores[i]) for i in range(2)}
 
 #examples = [[example_img.jpg], [example_img2.jpg]]  # must be uploaded in repo