Upload folder using huggingface_hub

app.py

@@ -63,7 +63,8 @@ def clear_file_fn():
     return None, None
 
 
-def retrieve_document_fn(question, instruct, document_data, document_index):
+def retrieve_document_fn(question, instruct, document_states):
+    document_data, document_index = document_states
     num_retrieval_doc = 3
     if document_index is None or document_data is None:
         gr.Warning("Please upload documents first!")
@@ -73,7 +74,7 @@ def retrieve_document_fn(question, instruct, document_data, document_index):
     batch_scores, batch_inxs = document_index.search(question_embedding, k=num_retrieval_doc)
     
     answers = [document_data[i]["text"] for i in batch_inxs[0][:num_retrieval_doc]]
-    return answers[0], answers[1], answers[2], document_data, document_index
+    return tuple(answers)
 
 
 def main(args):
@@ -98,8 +99,8 @@ def main(args):
         retrieval_interface = gr.Interface(
             fn=retrieve_document_fn,
             inputs=["text"],
-            outputs=["text", "text",  "text", gr.State(), gr.State()],
-            additional_inputs=[gr.Textbox("Instruct: Given a query, retrieve documents that answer the query. \n Query: ", label="Instruct of Query"), document_data, document_index],
+            outputs=["text", "text",  "text"],
+            additional_inputs=[gr.Textbox("Instruct: Given a query, retrieve documents that answer the query. \n Query: ", label="Instruct of Query"), [document_data, document_index]],
             concurrency_limit=1,
         )