Tong commited on
Commit
2e22a78
·
1 Parent(s): 91bccac

update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -1,9 +1,19 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModel
3
 
4
- def chat(text):
 
5
  global model, tokenizer
6
- response, history = model.chat(tokenizer, text, "", 64, 0.7, 0.95)
 
 
 
 
 
 
 
 
 
7
  return response
8
 
9
 
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModel
3
 
4
+ def chat(prompt):
5
+
6
  global model, tokenizer
7
+ response, history = model.chat(tokenizer,
8
+ prompt,
9
+ history=None,
10
+ max_length=64,
11
+ top_p=0.7,
12
+ temperature=0.95,
13
+ num_beam_groups=1,
14
+ num_beams=2
15
+ )
16
+
17
  return response
18
 
19