fCola commited on
Commit
cb15932
·
verified ·
1 Parent(s): 30e24b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -455,7 +455,7 @@ def send_message(message, history):
455
  "repetition_penalty": 1.25,
456
  }
457
  partial = ""
458
- thread = Thread(target=pipeline.generate, kwargs=gen_kwargs)
459
  thread.start()
460
  #for token in generator(message, max_new_tokens=200):
461
  for t in streamer:
 
455
  "repetition_penalty": 1.25,
456
  }
457
  partial = ""
458
+ thread = Thread(target=pipeline, kwargs=gen_kwargs)
459
  thread.start()
460
  #for token in generator(message, max_new_tokens=200):
461
  for t in streamer: