Update app.py

app.py

@@ -1,4 +1,19 @@
-#
+import os
+os.system("""
+# 1. safetensors -> PyTorch 形式に変換
+python -c "
+from transformers import AutoModelForCausalLM
+import torch
+model = AutoModelForCausalLM.from_pretrained('deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct')
+torch.save(model.state_dict(), 'pytorch_model.bin')
+"
+
+# 2. GGUF 形式に変換 (llama.cpp が必要)
+git clone https://github.com/ggerganov/llama.cpp
+cd llama.cpp
+python convert.py --outtype f16 /path/to/pytorch_model.bin
+
+""")
 from flask import Flask, request, Response, render_template
 from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
@@ -197,8 +212,8 @@ HTML_CONTENT = '''
 '''
 
 def download_model():
-    model_name = "lmstudio-community/gemma-2-2b-it-GGUF"
-    model_file = "gemma-2-2b-it-Q6_K.gguf"
+    model_name = "deepseek-ai/DeepSeek-Coder-V2-Lite-Instruct"
+    model_file = "deepseek-coder-v2-lite-instruct.Q6_K.gguf"  # 適切なGGUFファイル名に変更
     return hf_hub_download(model_name, filename=model_file)
 
 def initialize_model(model_path):
@@ -213,7 +228,8 @@ model_path = download_model()
 llm = initialize_model(model_path)
 
 system_prompt = (
-    "You are a normal AI assistant. Your mission is to help people and respond clearly and friendly."
+    "You are a helpful AI coding assistant. Your mission is to help people with programming "
+    "and technical questions, providing clear and concise answers."
 )
 
 chat_history = [{"role": "system", "content": system_prompt}]
@@ -246,4 +262,4 @@ def chat():
     return Response(generate(), content_type='text/event-stream')
 
 if __name__ == '__main__':
-    app.run(debug=True, port=7860, host="0.0.0.0")
+    app.run(debug=True, port=7860, host="0.0.0.0")