Update app.py

app.py

@@ -1,18 +1,12 @@
 # -*- coding: utf-8 -*-
-"""Hugging Face.ipynb
+"""Hugging Face Llama 2 App"""
 
-Automatically generated by Colab.
-
-Original file is located at
-    https://colab.research.google.com/drive/1zRuAxGm_11lNIeBxFlHVzc5tNKhyLef4
-"""
 import gradio as gr
-# 加載 LLaMA 模型
 from transformers import AutoTokenizer, AutoModelForCausalLM
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-tokenizer = AutoTokenizer.from_pretrained("google/t5-v1_1-xxl")
-model = AutoModelForSeq2SeqLM.from_pretrained("google/t5-v1_1-xxl")
+# 加載 LLaMA-2 模型
+tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-13b-chat-hf")
+model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-13b-chat-hf")
 
 # 定義推理函數
 def generate_text(prompt):
@@ -26,15 +20,15 @@ def generate_text(prompt):
     )
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-# 使用 Gradio 构建界面
+# 使用 Gradio 構建界面
 interface = gr.Interface(
     fn=generate_text,
     inputs=gr.Textbox(lines=5, placeholder="Enter your prompt here..."),
     outputs="text",
-    title="T5 Text Generator",
-    description="Generate text using the T5 v1.1 XXL model hosted on Hugging Face Spaces."
+    title="Llama 2 Text Generator",
+    description="Generate text using the Llama-2-13b-chat-hf model hosted on Hugging Face Spaces."
 )
 
 # 啟動應用
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()