已经完成了 xihua_model.pth 的训练,我们现在只需要编写一个使用该模型的聊天机器人羲和的代码。以下是完整的代码,包括加载模型、创建GUI界面以及处理用户输入和生成回答的功能。
聊天机器人羲和的代码
python
import os
import json
import jsonlines
import torch
from transformers import BertModel, BertTokenizer
import tkinter as tk
from tkinter import filedialog, messagebox
import logging# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')# 获取项目根目录
PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))# 数据集类(用于加载数据)
class XihuaDataset:def __init__(self, file_path):self.data = self.load_data(file_path)def load_data(self, file_path):data = []if file_path.endswith('.jsonl'):with jsonlines.open(file_path) as reader:for item in reader:data.append(item)elif file_path.endswith('.json'):with open(file_path, 'r') as f:data = json.load(f)return data# 模型定义
class XihuaModel