Update handler.py

handler.py

@@ -53,3 +53,5 @@ class EndpointHandler:
             del inputs
             gc.collect()
             torch.cuda.empty_cache()
+
+            return {"error": str(e)}