hayas commited on
Commit
7850ec9
1 Parent(s): ec4e41c
Files changed (3) hide show
  1. README.md +2 -2
  2. app.py +1 -1
  3. requirements.txt +2 -2
README.md CHANGED
@@ -1,10 +1,10 @@
1
  ---
2
- title: rinna Youri-7B
3
  emoji: ⚡
4
  colorFrom: red
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: 3.50.2
8
  app_file: app.py
9
  pinned: false
10
  license: mit
 
1
  ---
2
+ title: rinna Youri-7B-chat
3
  emoji: ⚡
4
  colorFrom: red
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: 4.0.2
8
  app_file: app.py
9
  pinned: false
10
  license: mit
app.py CHANGED
@@ -9,7 +9,7 @@ import spaces
9
  import torch
10
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
11
 
12
- DESCRIPTION = "# Youri-7B"
13
 
14
  if not torch.cuda.is_available():
15
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
 
9
  import torch
10
  from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
11
 
12
+ DESCRIPTION = "# Youri-7B-chat"
13
 
14
  if not torch.cuda.is_available():
15
  DESCRIPTION += "\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>"
requirements.txt CHANGED
@@ -1,9 +1,9 @@
1
  accelerate==0.24.1
2
  bitsandbytes==0.41.1
3
- gradio==3.50.2
4
  protobuf==3.20.3
5
  scipy==1.11.3
6
  sentencepiece==0.1.99
7
  spaces==0.18.0
8
  torch==2.0.0
9
- transformers==4.34.1
 
1
  accelerate==0.24.1
2
  bitsandbytes==0.41.1
3
+ gradio==4.0.2
4
  protobuf==3.20.3
5
  scipy==1.11.3
6
  sentencepiece==0.1.99
7
  spaces==0.18.0
8
  torch==2.0.0
9
+ transformers==4.35.0