codys12 commited on
Commit
0c42252
1 Parent(s): 73d0fad

remove hist 2

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -64,7 +64,7 @@ def generate(
64
  #top_k=top_k,
65
  #temperature=temperature,
66
  #num_beams=1,
67
- #repetition_penalty=repetition_penalty,
68
  )
69
  t = Thread(target=model.generate, kwargs=generate_kwargs)
70
  t.start()
@@ -72,9 +72,9 @@ def generate(
72
 
73
  outputs = []
74
  for text in streamer:
75
- print(text.replace(history, ""), end="")
76
  outputs.append(text)
77
- yield "".join(outputs)
78
 
79
 
80
  chat_interface = gr.ChatInterface(
 
64
  #top_k=top_k,
65
  #temperature=temperature,
66
  #num_beams=1,
67
+ repetition_penalty=0.0,#repetition_penalty,
68
  )
69
  t = Thread(target=model.generate, kwargs=generate_kwargs)
70
  t.start()
 
72
 
73
  outputs = []
74
  for text in streamer:
75
+ print(text, end="")
76
  outputs.append(text)
77
+ yield "".join(outputs).replace(history, "")
78
 
79
 
80
  chat_interface = gr.ChatInterface(