Update app.py

app.py

@@ -11,13 +11,17 @@ model_dir = "models/Miwa-Keita/zenz-v2.5-medium"
 snapshot_download(
     repo_id="Miwa-Keita/zenz-v2.5-medium",
     local_dir=model_dir,
-    allow_patterns=["*.bin", "*.json", "*.txt", "*.model"],  # 必要なファイルだけ取得
+    allow_patterns=["*.safetensors", "*.json", "*.txt", "*.model"],  # SafeTensorsに変更
     ignore_patterns=["optimizer.pt", "checkpoint*"],  # いらないファイルを無視
 )
 
 # モデルとトークナイザーのロード（GPT-2 アーキテクチャ）
 tokenizer = AutoTokenizer.from_pretrained(model_dir)
-model = AutoModelForCausalLM.from_pretrained(model_dir, torch_dtype=torch.float32)
+model = AutoModelForCausalLM.from_pretrained(
+    model_dir, 
+    torch_dtype=torch.float32, 
+    safe_serialization=True  # SafeTensorsを有効化
+)
 
 # 入力を調整する関数
 def preprocess_input(user_input):