Update app.py

app.py

@@ -2,11 +2,17 @@ import os
 import gradio as gr
 from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 
+messages = [
+    {"role": "user", "content": "Who are you?"},
+]
+pipe = pipeline("text-generation", model="SakanaAI/DiscoPOP-zephyr-7b-gemma")
+pipe(messages)
+
 # CUDAを無効にする
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
 # 日本語モデルを指定
-model_name = "sakanaai/evolllm-jp"  # Sakana AIのモデル名を指定
+model_name = "SakanaAI/DiscoPOP-zephyr-7b-gemma"  # Sakana AIのモデル名を指定
 
 # トークナイザーとパイプラインの設定
 tokenizer = AutoTokenizer.from_pretrained(model_name)