add main file

main.py

@@ -38,11 +38,9 @@ qa_chain = RetrievalQA.from_chain_type(llm=local_llm,
                                        chain_type="map_reduce",
                                        retriever=retriever,
                                        return_source_documents=True)
-
-question = input('prompt: ')
-result = qa_chain({'query': question})
-print('result: ', result['result'])
-
+# question = input('prompt: ')
+# result = qa_chain({'query': question})
+# print('result: ', result['result'])
 def gradinterface(query):
     result = qa_chain({'query': query})
     return result['result']
@@ -51,4 +49,4 @@ def gradinterface(query):
 demo = gr.ChatInterface(fn=gradinterface, title='OUR_OWN_BOT')
 
 if __name__ == "__main__":
-    demo.launch(show_api=False, share=True)
+    demo.launch(share=True)

requirements.txt

@@ -0,0 +1,5 @@
+langchain
+tiktoken
+chromadb
+transformers
+gradio