kelvinyasu commited on
Commit
114aae8
1 Parent(s): 870b2f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ headers = {
10
  }
11
 
12
  system_message = "\nTesting by KelvinLo UD\n"
13
- title = "Llama2 70B Chatbot"
14
  description = """
15
  Demo by Kelvin Lo, UD
16
  """
@@ -32,7 +32,7 @@ def predict(message, chatbot):
32
 
33
  data = {
34
  "inputs": input_prompt,
35
- "parameters": {"max_new_tokens":2560,
36
  "do_sample":True,
37
  "top_p":0.6,
38
  "temperature":0.9,}
 
10
  }
11
 
12
  system_message = "\nTesting by KelvinLo UD\n"
13
+ title = "Llama-2 Chatbot"
14
  description = """
15
  Demo by Kelvin Lo, UD
16
  """
 
32
 
33
  data = {
34
  "inputs": input_prompt,
35
+ "parameters": {"max_new_tokens": 5000,
36
  "do_sample":True,
37
  "top_p":0.6,
38
  "temperature":0.9,}