Jeff Chen commited on
Commit
637faa2
1 Parent(s): 887a14f

Change deps for GPU (cuda) problem

Browse files
Files changed (3) hide show
  1. README.md +4 -4
  2. app.py → main.py +0 -0
  3. requirements.txt +2 -1
README.md CHANGED
@@ -1,11 +1,11 @@
1
  ---
2
  title: Ckip-joint Bloom Streamlit
3
- emoji: 🐨
4
- colorFrom: gray
5
- colorTo: red
6
  sdk: streamlit
7
  sdk_version: 1.17.0
8
- app_file: app.py
9
  pinned: false
10
  license: mit
11
  ---
 
1
  ---
2
  title: Ckip-joint Bloom Streamlit
3
+ emoji: 🤩
4
+ colorFrom: blue
5
+ colorTo: green
6
  sdk: streamlit
7
  sdk_version: 1.17.0
8
+ app_file: main.py
9
  pinned: false
10
  license: mit
11
  ---
app.py → main.py RENAMED
File without changes
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
- torch==1.10.2+cu111
 
2
  transformers
3
  streamlit
4
 
 
1
+ # torch==1.10.2+cu111
2
+ torch
3
  transformers
4
  streamlit
5