Update app.py

app.py

@@ -69,10 +69,9 @@ carTransforms = transforms.Compose([transforms.Resize(224),
 def classifyCar(im):
   #im = Image.fromarray(im.astype('uint8'), 'RGB')
   im = cv2.imread(im)
-  with torch.no_grad():
-    outputs = predictor(im)
+  outputs = predictor(im)
   v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
-  out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
+  out = v.draw_instance_predictions(outputs["instances"])
     #im2 = carTransforms(im).unsqueeze(0)  # transform and add batch dimension
     #with torch.no_grad():
     #  scores = torch.nn.functional.softmax(DesignModernityModel(im2)[0])