| from transformers import pipeline, set_seed | |
| import streamlit as st | |
| generator = pipeline('text-generation', model='gpt2') | |
| set_seed(42) | |
| # 引入后台代码 | |
| def backend(question): | |
| # 在这里编写后台处理代码，此处为示例 | |
| res = generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5) | |
| return res[0] | |
| # 构建页面布局 | |
| st.title("GPT-2 Demo") | |
| question = st.text_input("请输入问题") | |
| if st.button("获取答案"): | |
| # 调用后台代码 | |
| answer = backend(question) | |
| st.write(answer) | |