update app

app.py

@@ -72,7 +72,7 @@ def load_model_from_config(config, ckpt, verbose=False):
         print("unexpected keys:")
         print(u)
 
-    model.cuda()
+    model = model.to(device)
     model.eval()
     return model