Update app.py

app.py

@@ -27,14 +27,8 @@ MODEL_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "models")
 # モデルディレクトリが存在しない場合は作成
 if not os.path.exists("models"):
     os.makedirs("models")
-
-# 使用するモデルのファイル名を指定
-model_filename = "Mistral-Nemo-Instruct-2407-Q8_0.gguf"
-model_path = os.path.join("models", model_filename) 
-
-# モデルファイルが存在しない場合はダウンロード
-if not os.path.exists(model_path):
-    dl_guff_model("models", f"https://huggingface.co/second-state/Mistral-Nemo-Instruct-2407-GGUF/resolve/main/{model_filename}")
+    dl_guff_model("models", "https://huggingface.co/MCZK/Llama-3.1-8B-EZO-1.1-it-GGUF/resolve/main/resolve/main/Llama-3.1-8B-EZO-1.1-it.Q8_0.gguf")
+    dl_guff_model("models", "https://huggingface.co/second-state/Mistral-Nemo-Instruct-2407-GGUF/resolve/main/Mistral-Nemo-Instruct-2407-Q8_0.gguf")
 
 
 class ConfigManager:
@@ -174,8 +168,8 @@ class Settings:
             "assistant: プロットについてコメントをする前に、まずこの物語の『売り』について簡単に説明してください",
             ],
             'gen_author_description': 'あなたは新進気鋭の和風伝奇ミステリー小説家で、細やかな筆致と巧みな構成で若い世代にとても人気があります。',
-            'DEFAULT_CHAT_MODEL': 'Mistral-Nemo-Instruct-2407-Q8_0.gguf',
-            'DEFAULT_GEN_MODEL': 'Mistral-Nemo-Instruct-2407-Q8_0.gguf',
+            'DEFAULT_CHAT_MODEL': 'Llama-3.1-8B-EZO-1.1-it.Q8_0.gguf',
+            'DEFAULT_GEN_MODEL': 'Llama-3.1-8B-EZO-1.1-it.Q8_0.gguf',
             'chat_n_gpu_layers': -1,
             'chat_temperature': 0.35,
             'chat_top_p': 0.9,
@@ -669,6 +663,13 @@ def build_gradio_interface():
         return f"{section}セクションの{key}を更新しました。適用ボタンを押すと設定が保存されます。"
 
     with gr.Blocks() as iface:
+        # 新しいメッセージを追加
+        gr.HTML("""
+        <div style="background-color: #f0f0f0; padding: 10px; margin-bottom: 10px; border-radius: 5px;">
+            <strong>注意：</strong> NSFW創作用途の場合はモデルを設定タブから、「Mistral-Nemo-Instruct-2407-Q8_0.gguf」に変更推奨です
+        </div>
+        """)
+        
         gr.HTML("""
         <style>
         #chatbot, #chatbot_read {