mobinln commited on
Commit
1c58dec
·
verified ·
1 Parent(s): 9a28b27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -46,7 +46,14 @@ def respond(
46
  print("messages", messages)
47
  output = ""
48
  for chunk in stream:
49
- delta = chunk.choices[0].delta.content or chunk.choices[0].delta.reasoning_content or ""
 
 
 
 
 
 
 
50
  output += delta
51
  yield output
52
 
@@ -57,7 +64,7 @@ def respond(
57
  demo = gr.ChatInterface(
58
  respond,
59
  additional_inputs=[
60
- gr.Textbox(value=".", label="System message"),
61
  gr.Slider(minimum=1, maximum=8000, value=4096, step=1, label="Max new tokens"),
62
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
63
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
 
46
  print("messages", messages)
47
  output = ""
48
  for chunk in stream:
49
+ delta = chunk.choices[0].delta
50
+ if "content" in delta:
51
+ delta = delta.content or chunk.choices[0].delta.reasoning_content
52
+ elif "reasoning_content" in delta:
53
+ delta = delta.reasoning_content
54
+ else:
55
+ continue
56
+
57
  output += delta
58
  yield output
59
 
 
64
  demo = gr.ChatInterface(
65
  respond,
66
  additional_inputs=[
67
+ gr.Textbox(value="", label="System message"),
68
  gr.Slider(minimum=1, maximum=8000, value=4096, step=1, label="Max new tokens"),
69
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
70
  gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),