RaynWu2002 commited on
Commit
fe86211
·
verified ·
1 Parent(s): bd4de0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import numpy as np
3
- import spaces
4
  import torch
5
  import os
6
  import cv2
@@ -21,12 +21,12 @@ import subprocess
21
  # print("Failed to install mmcv-full:", e)
22
 
23
  # install_mmcv()
24
- print("=" * 50)
25
- print(f"CUDA available: {torch.cuda.is_available()}")
26
- print(f"torch version: {torch.__version__}")
27
- print(f"CUDA version: {torch.version.cuda}")
28
- print(f"mmcv version: {mmcv.__version__}")
29
- print("=" * 50)
30
 
31
  # init
32
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
1
  import gradio as gr
2
  import numpy as np
3
+ # import spaces
4
  import torch
5
  import os
6
  import cv2
 
21
  # print("Failed to install mmcv-full:", e)
22
 
23
  # install_mmcv()
24
+ # print("=" * 50)
25
+ # print(f"CUDA available: {torch.cuda.is_available()}")
26
+ # print(f"torch version: {torch.__version__}")
27
+ # print(f"CUDA version: {torch.version.cuda}")
28
+ # print(f"mmcv version: {mmcv.__version__}")
29
+ # print("=" * 50)
30
 
31
  # init
32
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")