Update train.py

train.py

@@ -1,6 +1,7 @@
 import os
 import json
 import torch
+import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, Trainer, TrainingArguments
 from datasets import Dataset
 
@@ -156,3 +157,25 @@ trainer.train()
 # **保存模型**
 student.push_to_hub("Snow2222/fst-nnn", use_auth_token=hf_token)
 tokenizer.push_to_hub("Snow2222/fst-nnn", use_auth_token=hf_token)
+
+# ✅ 部署 Gradio Web 界面
+print("🎉 训练完成，启动 Gradio Web 界面...")
+
+# 加载微调后的模型
+pipe = pipeline("text-generation", model="Snow2222/fst-nnn", tokenizer="Snow2222/fst-nnn")
+
+# 定义 Gradio 交互函数
+def chat_response(prompt):
+    result = pipe(prompt, max_length=200)[0]["generated_text"]
+    return result
+
+# 启动 Gradio Web 界面
+iface = gr.Interface(
+    fn=chat_response,
+    inputs="text",
+    outputs="text",
+    title="AI 训练模型测试",
+    description="输入你的问题，模型将会自动回复！"
+)
+
+iface.launch(server_name="0.0.0.0", server_port=7860)