add partial

.ipynb_checkpoints/app-checkpoint.py

@@ -21,6 +21,7 @@ from scipy.integrate import odeint
 import torchvision
 import echonet
 import matplotlib.animation as animation
+from functools import partial
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 

app.py

@@ -21,6 +21,7 @@ from scipy.integrate import odeint
 import torchvision
 import echonet
 import matplotlib.animation as animation
+from functools import partial
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")