Update app.py

app.py

@@ -69,10 +69,16 @@ def classifyCar(im):
   try:
     with torch.no_grad():
       outputs = predictor(im)
+  except:
+    return im, {"error1": 0.5}
+  try:
       v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
+  except Exception as err:
+    return im, {"error2": 0.5}
+  try:
       out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
   except Exception as err:
-    return im, {err: float(0.5)}
+    return im, {"error3": 0.5}
   im2 = carTransforms(im).unsqueeze(0)  # transform and add batch dimension
   with torch.no_grad():
     scores = torch.nn.functional.softmax(DesignModernityModel(im2)[0])