Ensure README contains reference to gradio

README.md

@@ -3,8 +3,8 @@ title: Document Answering
 emoji: 🔥
 colorFrom: gray
 colorTo: purple
-sdk: streamlit
-sdk_version: 1.30.0
+sdk: gradio
+sdk_version: 3.1.7
 app_file: app.py
 pinned: false
 license: apache-2.0

app.py

@@ -1,3 +1,5 @@
+import time
+
 import gradio as gr
 from langchain.docstore.document import Document
 from langchain.text_splitter import RecursiveCharacterTextSplitter, Language
@@ -72,10 +74,14 @@ def predict(message, history):
     # resp = llm_model.answer_question_inference(message)
     # return resp.get("answer")
     resp = llm_model.answer_question_inference_text_gen(message)
-    final_resp = ""
-    for c in resp:
-        final_resp += str(c)
-        yield final_resp
+    for i in range(len(resp)):
+        time.sleep(0.005)
+        yield resp[:i + 1]
+    # final_resp = ""
+    # for c in resp:
+    #     final_resp += str(c)
+    #     # + "▌"
+    #     yield final_resp
     # start_time = time.time()
     # res = llm_model({"query": message})
     # sources = []
@@ -110,8 +116,8 @@ chat_interface_stream = gr.ChatInterface(
     description="📚🔦 Upload some documents on the side and ask questions!",
     textbox=gr.Textbox(container=False, scale=7),
     chatbot=chatbot_stream,
-    examples=["What is Data Caterer?", "Provide a set of potential questions and answers about the README"]
-)
+    examples=["What is Data Caterer?"]
+).queue(default_concurrency_limit=1)
 
 with gr.Blocks() as blocks:
     with gr.Row():