kz209 commited on
Commit
19083c7
·
1 Parent(s): ab46b56
Files changed (1) hide show
  1. utils/multiple_stream.py +4 -4
utils/multiple_stream.py CHANGED
@@ -28,10 +28,10 @@ def stream_data(content_list, model):
28
  updated = False
29
  #for i, content in enumerate(content_list):
30
  try:
31
- words = next(model.gen(content_list, streaming=True)) # Wrap content in a list to match expected input type
32
- print(outputs)
33
- print(words)
34
- outputs = [outputs[i]+f" {words[i]}" for i in range(len(content_list))]
35
  updated = True
36
  except StopIteration:
37
  pass
 
28
  updated = False
29
  #for i, content in enumerate(content_list):
30
  try:
31
+ id, words = next(model.gen(content_list, streaming=True)) # Wrap content in a list to match expected input type
32
+ # print(outputs)
33
+ # print(words)
34
+ outputs[id] += f" {words}"
35
  updated = True
36
  except StopIteration:
37
  pass