Julien Simon commited on
Commit
2eb5044
1 Parent(s): 82abe24

Install natten from url

Browse files
Files changed (2) hide show
  1. app.py +0 -10
  2. requirements.txt +1 -0
app.py CHANGED
@@ -1,16 +1,6 @@
1
- import subprocess
2
- import sys
3
-
4
  import gradio as gr
5
  from transformers import pipeline
6
 
7
-
8
- def install(package, index):
9
- subprocess.check_call([sys.executable, "-m", "pip", "install", package, index])
10
-
11
-
12
- install("natten", "-f https://shi-labs.com/natten/wheels/cpu/torch1.13/index.html")
13
-
14
  model_names = [
15
  "facebook/deit-base-patch16-224",
16
  "facebook/convnext-base-224",
 
 
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
 
 
 
 
 
 
 
4
  model_names = [
5
  "facebook/deit-base-patch16-224",
6
  "facebook/convnext-base-224",
requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  torch==1.13.1
2
  transformers>=4.25.1
 
 
1
  torch==1.13.1
2
  transformers>=4.25.1
3
+ https://shi-labs.com/natten/wheels/cpu/torch1.13/natten-0.14.4%2Btorch1130cpu-cp38-cp38-linux_x86_64.whl