PDG commited on
Commit
e8d20c7
·
1 Parent(s): ce9caa8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -66,11 +66,12 @@ carTransforms = transforms.Compose([transforms.Resize(224),
66
 
67
  def classifyCar(im):
68
  im = Image.fromarray(im.astype('uint8'), 'RGB')
 
69
  try:
70
  with torch.no_grad():
71
  outputs = predictor(im)
72
  except:
73
- return im, {"error1": 0.5}
74
  try:
75
  v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
76
  except Exception as err:
 
66
 
67
  def classifyCar(im):
68
  im = Image.fromarray(im.astype('uint8'), 'RGB')
69
+ im = transforms.ToTensor(im)
70
  try:
71
  with torch.no_grad():
72
  outputs = predictor(im)
73
  except:
74
+ return im, {"error1": im.shape}
75
  try:
76
  v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
77
  except Exception as err: