Update app.py

app.py

@@ -7,13 +7,13 @@ Original file is located at
     https://colab.research.google.com/drive/1zRuAxGm_11lNIeBxFlHVzc5tNKhyLef4
 """
 import gradio as gr
-
+import os
 # 加載 LLaMA 模型
 from transformers import AutoTokenizer, AutoModelForCausalLM
 token = os.getenv("Git Access")
 
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-hf",use_auth_token=token)
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-hf",use_auth_token=token)
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b-chat-hf",use_auth_token=token)
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b-chat-hf",use_auth_token=token)
 
 # 定義推理函數
 def generate_text(prompt):