vaishnav commited on
Commit
f2cbb22
·
1 Parent(s): faa8f77

implement streaming

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -5,6 +5,7 @@ import services.scraper
5
  import stores.chroma
6
  from llm_setup.llm_setup import LLMService
7
  from caching.lfu import LFUCache
 
8
 
9
  logger = logging.getLogger() # Create a logger object
10
  logger.setLevel(logging.INFO) # Set the logging level to INFO
@@ -34,7 +35,10 @@ def respond(user_input,session_hash):
34
  def echo(text, chat_history, request: gr.Request):
35
  if request:
36
  session_hash = request.session_hash
37
- return respond(text, session_hash)
 
 
 
38
  else:
39
  return "No request object received."
40
 
 
5
  import stores.chroma
6
  from llm_setup.llm_setup import LLMService
7
  from caching.lfu import LFUCache
8
+ import time
9
 
10
  logger = logging.getLogger() # Create a logger object
11
  logger.setLevel(logging.INFO) # Set the logging level to INFO
 
35
  def echo(text, chat_history, request: gr.Request):
36
  if request:
37
  session_hash = request.session_hash
38
+ resp = respond(text, session_hash)
39
+ for i in range(len(resp)):
40
+ time.sleep(0.05)
41
+ yield resp[: i + 1]
42
  else:
43
  return "No request object received."
44