fix _curses.error: setupterm: could not find terminal

app.py

@@ -13,6 +13,10 @@ from dotmap import DotMap
 from PIL import Image
 
 
+os.environ['TERM'] = 'linux'
+os.environ['TERMINFO'] = '/etc/terminfo'
+
+
 # args
 args = DotMap()
 args.deploy = 'vanilla'
@@ -24,7 +28,7 @@ args.cx = '06d75168141bc47f1'
 
 
 # model
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+device = 'cpu' #torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = get_model(args)
 model.to(device)
 #checkpoint = torch.load(args.resume, map_location='cpu')