PKU Hole Generator

Introduction

This is a course project for the NLP class of Introduction to AI in PKU.

This project aims to simulate the hole (树洞) in PKU. Based on the user's input, the AI models will generate replies that are similar to the real hole replies.

Environment

• python==3.9.6
• torch==1.11.0
• gensim==4.1.2
• numpy==1.21.5
• tensorboard==2.7.0
• streamlit==1.5.0

How to use

• Approach 1: visit this page.
• Approach 2: download zip and run streamlit run app.py.

Examples

😅

[Alice] 表白

[Bob] Bob

[Bob] 来了！

[Carol] 摸摸，因为要留到一个认真的日子！

[Bob] 谢谢A 爱你😘

popi

[Alice] 身高体重颜值

[洞主] 170，保密，自我感觉中上（会被人偶尔称赞的程度！）

[Bob] Re 洞主: dz是嘉心糖吗

[洞主] Re Bob: 确实

[Alice] Re 洞主: 或者想聊聊也行，我还挺会聊天的

---

For more details, check out our website or our presentation note.