Update app.py

app.py

@@ -12,7 +12,7 @@ from torch.nn.functional import sigmoid
 from collections import Counter
 import logging
 import traceback
-
+import atexit
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG)