update for jit cpu load

app.py

@@ -20,6 +20,7 @@ if torch.cuda.is_available():
         device = 'cuda'
 else:
     device = 'cpu'
+    torch._C._jit_set_bailout_depth(0)
 
 print('Use device:', device)