Update app.py

app.py

@@ -1391,7 +1391,13 @@ if __name__ == "__main__":
     parser.add_argument('-demo', '--demo_mode', action='store_true', help='Enable demo mode')
     #parser.add_argument('--log_file', action=str, help='Where to save logfile (non-default)')
     args = parser.parse_args()
+    
+    print(f"Is CUDA available: {torch.cuda.is_available()}")
+    # True
+    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+    # Tesla T4
 
+    
     # Since this is running in HF....
     args.user_interface = True
     if args.user_interface: