likewendy commited on
Commit
d76503c
·
1 Parent(s): 2d3030d
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -50,7 +50,7 @@ def respond(
50
  temperature=temperature,
51
  streamer=streamer,
52
  top_p=top_p,
53
- return_full_text=True,
54
  do_sample=True,
55
  )
56
 
@@ -60,9 +60,7 @@ def respond(
60
  thread.start()
61
  # Print the generated text in real-time
62
  for new_text in streamer:
63
- response += new_text
64
- response = tokenizer.decode(tokenizer.encode(response), skip_special_tokens=True)
65
- yield response
66
 
67
  with gr.Blocks() as demo:
68
  with gr.Row():
 
50
  temperature=temperature,
51
  streamer=streamer,
52
  top_p=top_p,
53
+ return_full_text=False,
54
  do_sample=True,
55
  )
56
 
 
60
  thread.start()
61
  # Print the generated text in real-time
62
  for new_text in streamer:
63
+ yield tokenizer.decode(tokenizer.encode(new_text), skip_special_tokens=True)
 
 
64
 
65
  with gr.Blocks() as demo:
66
  with gr.Row():