Skip to content

yuanzhoulvpi2017/DocumentSearch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

文档搜索

  1. 🥱懒得用langchain,看他文档,可能需要一天,我自己实现的话,一天都不需要。
  2. 💻 本质上就是sbertchatglm-6b
  3. 👨‍💻 自己做一个得了。
  4. 🗑️目前还是毛坯,后面继续更新。
  5. 🎯只要传递一个文件夹📁,就可以把这个文件夹下所有的.pdf.docx格式文件加载(目前还不支持的.doc格式文件)

用法

命令行形式

  1. demo.py文件里面
if __name__ == "__main__":
    global_dir = "政策归档文件" # 你自己的文件夹

    kl = KnowLedge(global_dir=global_dir)
    res, data = kl.search_result(question_str="大学生创业有什么补贴") # 你想问什么呢
    print(res)
    print(data)

web端

  1. 注意修改code-21行的代码global_dir = "政策归档文件"
streamlit run web_ui.py --server.fileWatcherType none

About

基于sentence transformers和chatglm实现的文档搜索工具

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages