Update app.py

app.py

@@ -8,8 +8,8 @@ import os
 # 加载 Whisper 模型
 whisper_model = whisper.load_model("base")
 
-# 使用 GPT-Neo 生成中文提示词
-model_name = "EleutherAI/gpt-neo-2.7B"  # 你可以选择其它的模型
+# 使用 DistilGPT-2 生成中文提示词
+model_name = "distilgpt2"  # DistilGPT-2 是一个较小的 GPT-2 版本
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
@@ -87,4 +87,3 @@ demo.launch()
 
 
 
-