MrYxJ commited on
Commit
28cab40
1 Parent(s): 7cbf317
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@
8
  Mail : yxj2017@gmail.com
9
  Github : https://github.com/MrYxJ
10
  Date : 2023-09-05 23:25:28
11
- LastEditTime : 2023-09-09 19:56:42
12
  Copyright (C) 2023 mryxj. All rights reserved.
13
  '''
14
  import gradio as gr
@@ -84,7 +84,7 @@ with gr.Blocks() as demo:
84
  show_legend=True, show_copy_button=True)
85
 
86
  with gr.Row():
87
- inp = gr.Textbox(label="Model Name or URL", value="baichuan-inc/Baichuan2-13B-Chat")
88
  access_token = gr.Textbox(label="API Token", placeholder="Optional (for gated models)")
89
 
90
  with gr.Row():
 
8
  Mail : yxj2017@gmail.com
9
  Github : https://github.com/MrYxJ
10
  Date : 2023-09-05 23:25:28
11
+ LastEditTime : 2023-09-09 20:07:54
12
  Copyright (C) 2023 mryxj. All rights reserved.
13
  '''
14
  import gradio as gr
 
84
  show_legend=True, show_copy_button=True)
85
 
86
  with gr.Row():
87
+ inp = gr.Textbox(label="Model Name or URL", value="bert-base-cased")
88
  access_token = gr.Textbox(label="API Token", placeholder="Optional (for gated models)")
89
 
90
  with gr.Row():
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
  calflops
2
- bitsandbytes
3
  accelerate @ git+https://github.com/huggingface/accelerate
4
  transformers
5
  timm
 
1
  calflops
2
+ bitsandbytes==0.41.1
3
  accelerate @ git+https://github.com/huggingface/accelerate
4
  transformers
5
  timm