mitulagr2 commited on
Commit
95d75bf
1 Parent(s): 1c8932d
Files changed (1) hide show
  1. app/main.py +2 -2
app/main.py CHANGED
@@ -31,9 +31,9 @@ logger = logging.getLogger(__name__)
31
  def astreamer(generator):
32
  t0 = time.time()
33
  for i in generator:
34
- logger.info(f"Chunk being yielded (time {int((time.time()-t0)*1000)}ms)", flush=True)
35
  yield i
36
- logger.info(f"Over (time {int((time.time()-t0)*1000)}ms)", flush=True)
37
 
38
 
39
  @app.get("/query")
 
31
  def astreamer(generator):
32
  t0 = time.time()
33
  for i in generator:
34
+ logger.info(f"Chunk being yielded (time {int((time.time()-t0)*1000)}ms)")
35
  yield i
36
+ logger.info(f"Over (time {int((time.time()-t0)*1000)}ms)")
37
 
38
 
39
  @app.get("/query")