Staticaliza commited on
Commit
da8de96
·
verified ·
1 Parent(s): 52ec652

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -62,6 +62,7 @@ def generate(prompt, temperature, top_p, top_k, repetition_penalty, max_tokens,
62
  if event.is_set():
63
  raise TimeoutError("[ERROR] Generation timed out.")
64
  buffer += item["choices"][0]["text"]
 
65
  yield buffer
66
  finally:
67
  timer.cancel()
 
62
  if event.is_set():
63
  raise TimeoutError("[ERROR] Generation timed out.")
64
  buffer += item["choices"][0]["text"]
65
+ print(item)
66
  yield buffer
67
  finally:
68
  timer.cancel()