app.py, Zmaker.py, requirements.txtのアップロード

app.py : streamlitによるGUI制御
Zmaker.py : fine-tuning済みのGPT-2で推論を行うためのコード

app.py

@@ -0,0 +1,94 @@
+import streamlit as st
+from Zmaker import Zmaker
+
+if __name__ == "__main__":
+
+    #ファインチューニング済みモデルの読み込み
+    with st.spinner(text = "loading GPT-2..."):
+        if not ("AI" in st.session_state.keys()):
+            st.session_state["AI"] = Zmaker(
+                ft_path = "model/gpt2-ft/"
+            )
+    
+    #設定用サイドバーの設定
+    with st.sidebar:
+        st.title("GPT-2のパラメータ")
+
+        #max_lenの設定用スライダ
+        sld_max_len = st.sidebar.slider(
+            "length of the sentence", min_value = 0, max_value = 256, 
+            value = (25, 75), step = 1, key = "length"
+        )
+
+        #temperatureの設定用スライダ
+        sld_temp = st.sidebar.slider(
+            "temperature", min_value = 0.1, max_value = 1.5, 
+            value = 0.1, step = 0.1, key = "temp"
+        )
+        
+        #top_kの設定用スライダ
+        sld_top_k = st.sidebar.slider(
+            "top_k", min_value = 0, max_value = 500, 
+            value = 40, step = 1, key = "top_k"
+        )
+        
+        #top_pの設定用スライダ
+        sld_top_p = st.sidebar.slider(
+            "top_p", min_value = 0.01, max_value = 1.0, 
+            value = 0.95, step = 0.01, key = "top_p"
+        )
+
+        #repeat_ngram_sizeの設定用スライダ
+        sld_top_p = st.sidebar.slider(
+            "repeat_ngram_size ", min_value = 1, max_value = 10, 
+            value = 1, step = 1, key = "repeat_ngram_size"
+        )
+
+    #メインフォームの設定
+    with st.form(key = "Letter Form", clear_on_submit = False):
+        st.title("おてがみ 入力欄")
+        body = st.empty()
+        if ("letter_body" in st.session_state.keys()): 
+            ret = body.text_area(
+                label = "お手紙を途中まで漢字+ひらがなで書いてください。続きをＡＩが生成します。\n"\
+                        "本アプリで生成できるのは本文のみです。",
+                value = st.session_state["letter_body"]
+            )
+        else:
+            ret = body.text_area(
+                label = "お手紙を途中まで漢字+ひらがなで書いてください。\n"\
+                        "続きをＡＩが生成します。",
+                value = "ズッポシ村へようこそ！"
+            )
+        sub = st.form_submit_button("Generate")
+    
+    #注意事項
+    with st.expander("注意事項"):
+        st.text(
+                "※このAIは「どうぶつの森e+実況プレイ」"\
+                "　(https://www.nicovideo.jp/mylist/45062007)において"\
+                "　稲葉百万鉄氏により作成された文章を学習データに用いております。\n"
+                "　また，教師データの作成においてmintmama氏の作成した"\
+                "　「ズッポシむら手紙集」(https://www.nicovideo.jp/series/85494)\n"\
+                "を用いております。"
+        )
+
+       
+    #submitボタンが押された
+    if sub == True:
+        #predictに必要な条件をGUIで設定した値に更新
+        st.session_state["AI"].min_len = st.session_state["length"][0]
+        st.session_state["AI"].max_len = st.session_state["length"][-1]
+        st.session_state["AI"].top_k = st.session_state["top_k"]
+        st.session_state["AI"].top_p = st.session_state["top_p"]
+        st.session_state["AI"].temp = st.session_state["temp"]
+        st.session_state["AI"].repeat_ngram_size = st.session_state["repeat_ngram_size"]
+
+        #AIによる予測を実行
+        with st.spinner(text = "generating..."):
+            prompt = ret
+            text  = str(st.session_state["AI"].GenLetter("<s>"+prompt)[0])
+            text = text.replace('<s>', '')
+            text = text.replace('</s>', '')
+            st.session_state["letter_body"] = text
+            st.experimental_rerun()

requirements.txt

@@ -0,0 +1,67 @@
+accelerate==0.18.0
+altair==4.2.2
+attrs==23.1.0
+blinker==1.6.2
+cachetools==5.3.0
+certifi==2022.12.7
+charset-normalizer==3.1.0
+click==8.1.3
+colorama==0.4.6
+decorator==5.1.1
+entrypoints==0.4
+filelock==3.12.0
+fsspec==2023.4.0
+gitdb==4.0.10
+GitPython==3.1.31
+huggingface-hub==0.14.1
+idna==3.4
+importlib-metadata==6.6.0
+Jinja2==3.1.2
+JsonForm==0.0.2
+jsonschema==4.17.3
+JsonSir==0.0.2
+markdown-it-py==2.2.0
+MarkupSafe==2.1.2
+mdurl==0.1.2
+mojimoji==0.0.12
+mpmath==1.3.0
+networkx==3.1
+numpy==1.24.3
+packaging==23.1
+pandas==2.0.1
+Pillow==9.5.0
+protobuf==3.20.3
+psutil==5.9.5
+pyarrow==12.0.0
+pydeck==0.8.1b0
+Pygments==2.15.1
+Pympler==1.0.1
+pyrsistent==0.19.3
+python-dateutil==2.8.2
+Python-EasyConfig==0.1.7
+pytz==2023.3
+pytz-deprecation-shim==0.1.0.post0
+PyYAML==6.0
+regex==2023.3.23
+requests==2.29.0
+rich==13.3.5
+sentencepiece==0.1.99
+six==1.16.0
+smmap==5.0.0
+streamlit==1.22.0
+sympy==1.11.1
+tenacity==8.2.2
+tokenizers==0.13.3
+toml==0.10.2
+toolz==0.12.0
+torch==2.0.0
+tornado==6.3.1
+tqdm==4.65.0
+transformers==4.28.1
+typing_extensions==4.5.0
+tzdata==2023.3
+tzlocal==4.3
+urllib3==1.26.15
+validators==0.20.0
+watchdog==3.0.0
+zipp==3.15.0