update

app.py

@@ -1,4 +1,40 @@
 import streamlit as st
+import time
+import requests
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+st.header("DeeplySorry")
+
+all_input = st.text_area('模型输入', value="""
+今天，我们正式发布名为 DeeplySorry 的大规模神经网络模型，它可以代替您向您珍惜的亲人、朋友、爱人道歉。\n""", height=100)
+
+max_tokens = st.slider('生成的最大长度', min_value=15, max_value=1024, value=64)
+
+
+def completion(prompt):
+    start = time.monotonic()
+    resp = requests.post('https://welm.weixin.qq.com/v1/completions', json={
+        'prompt': prompt,
+        'max_tokens': max_tokens,
+        'temperature': 0.85,
+        'top_p': 0.95,
+        'top_k': 0.0,
+        'n': 5,
+        'stop': None,
+        # 'stop': [[13, 13]],
+    }, headers={"Authorization": f"Bearer {st.secrets['token']}"})
+    answers = resp.json()
+    print(answers)
+    answers = [c['text'] for c in answers['choices'] if c['text'] is not None]
+    cols = st.columns(3)
+    for idx, answer in enumerate(answers):
+        if idx >= 3:
+            break
+        with cols[idx]:
+            content = (prompt + answer).replace("\n", "\n\n")
+            st.markdown(f'## 版本{idx}\n\n{content}')
+    end = time.monotonic()
+    st.text(f'耗时：{end - start}')
+
+
+if st.button('开始生成/换一批'):
+    completion(all_input)