Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,12 +1,12 @@
|
|
| 1 |
-
|
| 2 |
import gradio as gr
|
| 3 |
import torch
|
| 4 |
from torchvision import models, transforms
|
| 5 |
-
|
| 6 |
# -- get torch and cuda version
|
| 7 |
TORCH_VERSION = ".".join(torch.__version__.split(".")[:2])
|
| 8 |
CUDA_VERSION = torch.__version__.split("+")[-1]
|
| 9 |
-
|
| 10 |
# -- install pre-build detectron2
|
| 11 |
!pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/{CUDA_VERSION}/{TORCH_VERSION}/index.html
|
| 12 |
|
|
|
|
| 1 |
+
|
| 2 |
import gradio as gr
|
| 3 |
import torch
|
| 4 |
from torchvision import models, transforms
|
| 5 |
+
'''
|
| 6 |
# -- get torch and cuda version
|
| 7 |
TORCH_VERSION = ".".join(torch.__version__.split(".")[:2])
|
| 8 |
CUDA_VERSION = torch.__version__.split("+")[-1]
|
| 9 |
+
|
| 10 |
# -- install pre-build detectron2
|
| 11 |
!pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/{CUDA_VERSION}/{TORCH_VERSION}/index.html
|
| 12 |
|