Update app.py

app.py

@@ -86,7 +86,6 @@ def classifyCar(im):
     im2 = carTransforms(im2)
 
     label = "success"
-    img = Image.toarray(im2)
   except:
     label = "fail2"
     #im2 = carTransforms(im).unsqueeze(0)  # transform and add batch dimension
@@ -94,7 +93,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 label, img
+  return label, im2
 
 #examples = [[example_img.jpg], [example_img2.jpg]]  # must be uploaded in repo