PDG commited on
Commit
6b2d891
·
1 Parent(s): 108b092

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,12 +1,12 @@
1
- #!pip install torch==1.10.0
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