Update app.py

app.py

@@ -82,8 +82,11 @@ def classifyCar(im):
     
     im2 = Image.fromarray(np.uint8(im)).convert('RGB').crop(boxes[max_idx].to(torch.int64).numpy())
     
-    carTransforms = transforms.Compose([transforms.Resize((224, 224))])
-    im2 = carTransforms(im2)
+    carResize = transforms.Compose([transforms.Resize((224, 224))])
+    im2 = carResize(im2)
+    
+    with torch.no_grad():
+      scores = torch.nn.functional.softmax(DesignModernityModel(carTransforms(im2).unsqueeze(0))[0])
 
     label = "success"
   except:
@@ -93,7 +96,7 @@ def classifyCar(im):
     #  scores = torch.nn.functional.softmax(DesignModernityModel(im2)[0])
       #{LABELS[i]: float(scores[i]) for i in range(n_labels)}
       #Image.fromarray(np.uint8(out.get_image())).convert('RGB')
-  return im2, label
+  return im2, {LABELS[i]: float(scores[i]) for i in range(n_labels)}
 
 #examples = [[example_img.jpg], [example_img2.jpg]]  # must be uploaded in repo