Update app.py

app.py

@@ -7,10 +7,21 @@ from transformers import AutoTokenizer
 import ctranslate2
 import gradio as gr
 
+DESCRIPTION="""
+## 概要
+- これは、とある研究発表のために作られたチャットルーム(スペース)です。アクセス過多の場合は、少し時間をおいてから再度アクセスしてください。
+- 詳細設定にて、AIが生成する文章のテイストを調整することが出来ます。
+- AIの名前は「ベータ」です。
+## たのむぞ
+- あまり個人情報を入力しないでください。
+- 会話内容は収集しておりません。
+"""
+
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 generator = ctranslate2.Generator("./FixedStar-BETA-7b-ct2", device=device)
 tokenizer = AutoTokenizer.from_pretrained(
-    "Yumenohoshi/Fixedstar-BETA", use_fast=True)
+    "Yumenohoshi/Fixedstar-BETA-7b", use_fast=True)
 
 def inference_func(prompt, max_length=128, sampling_temperature=0.7):
     tokens = tokenizer.convert_ids_to_tokens(
@@ -52,8 +63,9 @@ def interact_func(message, chat_history, max_context_size, max_length, sampling_
     chat_history.append((message, generated))
     return "", chat_history
 
+gr.Markdown(DESCRIPTION)
 
-with gr.Blocks(theme="monochrome") as demo:
+with gr.Blocks(theme="monochrome", css="style.css") as demo:
     with gr.Accordion("Configs", open=False):
         # max_context_size = the number of turns * 2
         max_context_size = gr.Number(value=20, label="記憶する会話ターン数", precision=0)