Update app.py

app.py

@@ -20,9 +20,9 @@ import os
 import logging
 
 try:
-    os.system('nvidia-smi')
+    import cupy
 except:
-    print('get cuda information failed!')
+    print('import cupy failed!')
 
 import numpy as np
 import torch