ML610 commited on
Commit
8ed0ff8
1 Parent(s): b39fe03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ history = []
8
 
9
  def generateResponse(prompt, history):
10
  formattedPrompt = f"<s>[INST] {prompt} [/INST]"
11
- response = model(formattedPrompt, max_new_tokens=1024, threads=8)
12
  history.append([prompt, response])
13
  return response
14
 
@@ -28,5 +28,5 @@ UI = gr.ChatInterface(
28
  clear_btn="Clear chat"
29
  )
30
 
31
- UI.queue(max_size=10, concurrency_count=8)
32
  UI.launch()
 
8
 
9
  def generateResponse(prompt, history):
10
  formattedPrompt = f"<s>[INST] {prompt} [/INST]"
11
+ response = model(formattedPrompt, max_new_tokens=1024)
12
  history.append([prompt, response])
13
  return response
14
 
 
28
  clear_btn="Clear chat"
29
  )
30
 
31
+ UI.queue(max_size=10, concurrency_count=16)
32
  UI.launch()