louisbrulenaudet commited on
Commit
6ed5e3d
1 Parent(s): 08f58b6

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -85,7 +85,6 @@ model, tokenizer, description = setup(
85
  description=DESCRIPTION
86
  )
87
 
88
- print(model, tokenizer)
89
 
90
  def preprocess_conversation(
91
  message: str,
@@ -271,9 +270,9 @@ def generate(
271
  t.start()
272
 
273
  outputs = []
274
-
275
  for text in streamer:
276
  outputs.append(text)
 
277
 
278
  return "".join(outputs)
279
 
 
85
  description=DESCRIPTION
86
  )
87
 
 
88
 
89
  def preprocess_conversation(
90
  message: str,
 
270
  t.start()
271
 
272
  outputs = []
 
273
  for text in streamer:
274
  outputs.append(text)
275
+ yield "".join(outputs)
276
 
277
  return "".join(outputs)
278