from transformers import BertTokenizer, BertForSequenceClassification
import torch
def recognize_intent(text, model_name='bert-base-uncased', num_labels=3):
tokenizer = BertTokenizer.from_pretrained(model_name)
model = BertForSequenceClassification.from_pretrained(model_name, num_labels=num_labels)
inputs = tokenizer(text, return_tensors='pt', max_length=512, truncation=True, padding=True)
outputs = model(**inputs)
probs = torch.nn.functional.softmax(outputs.logits, dim=-1)
label = torch.argmax(probs, dim=-1).item()
return label
from transformers import GPT2LMHeadModel, GPT2Tokenizer
def generate_response(text, max_length=100, temperature=0.7, model_name='gpt2'):
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
model = GPT2LMHeadModel.from_pretrained(model_name)
inputs = tokenizer(text, return_tensors='pt', max_length=1024, truncation=True)
outputs = model.generate(
**inputs,
max_length=max_length,
num_beams=5,
early_stopping=True,
temperature=temperature
)
output_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
return output_text
import openai
def generate_response_gpt3(text, max_tokens=100, temperature=0.7):
openai.api_key = 'YOUR_API_KEY'
response = openai.Completion.create(
engine="text-davinci-003",
prompt=text,
max_tokens=max_tokens,
n=1,
stop=None,
temperature=temperature
)
generated_text = response.choices[0].text.strip()
return generated_text
import tkinter as tk
from tkinter import scrolledtext, messagebox
from transformers import GPT2LMHeadModel, GPT2Tokenizer
class TextInputFrame(tk.Frame):
def __init__(self, parent, on_process):
super().__init__(parent)
self.on_process = on_process
self.create_widgets()
def create_widgets(self):
self.text_input = scrolledtext.ScrolledText(self, width=60, height=10)
self.text_input.pack(pady=10, padx=10, fill="both", expand=True)
tk.Button(self, text="发送", command=self.process_text).pack(pady=10, padx=10)
def process_text(self):
text = self.text_input.get("1.0", tk.END)
if text.strip():
self.on_process(text.strip())
else:
messagebox.showwarning("警告", "请输入文本")
class ResultFrame(tk.Frame):
def __init__(self, parent):
super().__init__(parent)
self.create_widgets()
def create_widgets(self):
self.result_text = scrolledtext.ScrolledText(self, width=60, height=5)
self.result_text.pack(pady=10, padx=10, fill="both", expand=True)
def display_result(self, result):
self.result_text.delete("1.0", tk.END)
self.result_text.insert(tk.END, result)
def generate_response(text, max_length=100, temperature=0.7, model_name='gpt2'):
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
model = GPT2LMHeadModel.from_pretrained(model_name)
inputs = tokenizer(text, return_tensors='pt', max_length=1024, truncation=True)
outputs = model.generate(
**inputs,
max_length=max_length,
num_beams=5,
early_stopping=True,
temperature=temperature
)
return tokenizer.decode(outputs[0], skip_special_tokens=True)
class ChatbotApp:
def __init__(self, root):
self.root = root
self.root.title("客户服务聊天机器人应用")
self.model_loaded = False
self.create_widgets()
def create_widgets(self):
self.text_input_frame = TextInputFrame(self.root, self.process_text)
self.text_input_frame.pack(pady=10, padx=10, fill="both", expand=True)
self.result_frame = ResultFrame(self.root)
self.result_frame.pack(pady=10, padx=10, fill="both", expand=True)
def process_text(self, text):
try:
response = generate_response(text)
self.result_frame.display_result(response)
except Exception as e:
messagebox.showerror("错误", f"处理失败:{str(e)}")
if __name__ == "__main__":
root = tk.Tk()
app = ChatbotApp(root)
root.mainloop()