Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|