Spaces:
Running
Running
vaishnav commited on
Commit ·
f2cbb22
1
Parent(s): faa8f77
implement streaming
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
| 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 |
|