Update app.py

app.py

@@ -19,6 +19,11 @@ import argparse
 import os
 import logging
 
+try:
+    os.system('nvidia-smi')
+except:
+    print('get cuda information failed!')
+
 import numpy as np
 import torch
 from PIL import Image