|
|
from transformers import pipeline |
|
|
|
|
|
class SimpleAssistant: |
|
|
def __init__(self, model_name="facebook/bart-large-cnn"): |
|
|
""" |
|
|
สร้าง AI ผู้ช่วยอย่างง่ายโดยใช้โมเดลภาษาที่กำหนด |
|
|
""" |
|
|
self.model_name = model_name |
|
|
self.nlp = pipeline("text-generation", model=self.model_name) |
|
|
|
|
|
def get_response(self, user_input, max_length=50, num_return_sequences=1): |
|
|
""" |
|
|
สร้างการตอบสนองต่ออินพุตของผู้ใช้ |
|
|
""" |
|
|
try: |
|
|
response = self.nlp(user_input, max_length=max_length, num_return_sequences=num_return_sequences) |
|
|
return response[0]['generated_text'] |
|
|
except Exception as e: |
|
|
print(f"เกิดข้อผิดพลาด: {e}") |
|
|
return "ขออภัย ฉันไม่เข้าใจ" |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
assistant = SimpleAssistant() |
|
|
|
|
|
print("ผู้ช่วยพร้อมใช้งานแล้ว! พิมพ์ 'ออก' เพื่อจบการสนทนา") |
|
|
|
|
|
while True: |
|
|
user_input = input("คุณ: ") |
|
|
if user_input.lower() == "ออก": |
|
|
print("ลาก่อน!") |
|
|
break |
|
|
|
|
|
response = assistant.get_response(user_input) |
|
|
print("ผู้ช่วย: " + response) |
