Update app.py

app.py

@@ -7,13 +7,12 @@ 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")
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-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)
+tokenizer = AutoTokenizer.from_pretrained("google/t5-v1_1-xxl")
+model = AutoModelForSeq2SeqLM.from_pretrained("google/t5-v1_1-xxl")
 
 # 定義推理函數
 def generate_text(prompt):
@@ -32,8 +31,8 @@ interface = gr.Interface(
     fn=generate_text,
     inputs=gr.Textbox(lines=5, placeholder="Enter your prompt here..."),
     outputs="text",
-    title="LLaMA Text Generator",
-    description="Generate text using LLaMA 2 models hosted on Hugging Face Spaces."
+    title="T5 Text Generator",
+    description="Generate text using the T5 v1.1 XXL model hosted on Hugging Face Spaces."
 )
 
 # 啟動應用