写了个平时看论文可能会用上的工具

写了一款可以帮助经常阅读论文的同学提升效率的小工具,有两个部分:

  • HuggingFace Weekly Paper 制作工具
  • Arxiv Helper

前置条件

翻译基于 deepseek 的服务, 论文十问依赖于 kimi, 因此需要这两个的 api key。可以配置到环境变量中,代码中使用的环境变量 key 分别是

  • DEEPSEEK_KEY
  • KIMI_KEY

如果不想同时用两家,翻译可以考虑也换成 kimi,需要手动修改代码,将 deepseek 里面的设置换成 kimi的。

1. HuggingFace Weekly Paper 制作工具

我每周博客和公众号上都会发一篇 weekly paper,文章来源于 HuggingFace 的 Daily Paper。 为了减少每次都要一个个点进去通过N次复制粘贴来得到翻译后的结果的痛苦,写了个脚本,可以直接读取本周的点赞超过n次的论文,并生成Weekly Paper。

代码就是hf.py文件,运行 weekly_paper 方法即可,慢慢等待即可,如果出现了一些翻译上的问题或者接口异常,可以重新从目录下的output.md文件里面拿到英文原版继续人工处理。

此脚本依赖的模型是 deepseek.

2. Arxiv Helper

这个使用 streamlit 做了前端,输入框里面输入 arxiv 论文的首页地址,回车即可得到它的标题和摘要,左侧可以进行翻译,右侧是论文十问的模板,使用 kimi 的接口进行问答。

对pdf的读写是依赖于kimi的基于文件的问答,因此会上传文件到kimi,文件上传列表可以在设置页面看到(初版暂时没做文件上传去重处理),不需要的可以删除。文件本身也会下载到data目录,每篇论文对应一个文件夹,里面存放pdf以及数据json,避免每次都要重新调用接口。

使用方式:

shell 复制代码
streamlit run arxiv.py

后续计划

  1. arxiv helper 支持跳转到 kimi 聊天,并带上已经进行过的论文十问;
  2. 论文上传去重
  3. 本地论文管理(待定)

GitHub地址, 有用请点个 star,非常感谢!

相关推荐
数据与后端架构提升之路4 分钟前
TeleTron 源码揭秘:如何用适配器模式“无缝魔改” Megatron-Core?
人工智能·python·适配器模式
Chef_Chen35 分钟前
数据科学每日总结--Day44--机器学习
人工智能·机器学习
这张生成的图像能检测吗36 分钟前
(论文速读)FR-IQA:面向广义图像质量评价:放松完美参考质量假设
人工智能·计算机视觉·图像增强·图像质量评估指标
KG_LLM图谱增强大模型1 小时前
本体论与知识图谱:揭示语义技术的核心差异
人工智能·知识图谱·本体论
JicasdC123asd2 小时前
黄瓜植株目标检测:YOLOv8结合Fasternet与BiFPN的高效改进方案
人工智能·yolo·目标检测
爱吃泡芙的小白白2 小时前
深入解析:2024年AI大模型核心算法与应用全景
人工智能·算法·大模型算法
小程故事多_803 小时前
攻克RAG系统最后一公里 图文混排PDF解析的挑战与实战方案
人工智能·架构·pdf·aigc
琅琊榜首20203 小时前
AI+编程双驱动:高质量短剧创作全流程指南
人工智能
Master_oid4 小时前
机器学习29:增强式学习(Deep Reinforcement Learning)④
人工智能·学习·机器学习
Cemtery1164 小时前
Day26 常见的降维算法
人工智能·python·算法·机器学习