Update app.py

app.py

@@ -81,9 +81,15 @@ def classifyCar(im):
       out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
   except Exception as err:
     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])
+  try:
+    im2 = carTransforms(im).unsqueeze(0)  # transform and add batch dimension
+  except:
+    return im, {"error4": 0.5}
+  try:
+    with torch.no_grad():
+      scores = torch.nn.functional.softmax(DesignModernityModel(im2)[0])
+  except:
+    return im, {"error5": 0.5}
   return Image.fromarray(np.uint8(out.get_image())).convert('RGB'), {LABELS[i]: float(scores[i]) for i in range(n_labels)}
 
 #examples = [[example_img.jpg], [example_img2.jpg]]  # must be uploaded in repo