Ai-bocphet / README.md
HuyRemy's picture
Update README.md
9534677 verified

Open In Colab

from transformers import pipeline

model_name = "HuyRemy/Ai-bocphet"
chatbot = pipeline("text-generation", model=model_name)

def generate_response(prompt):
    response = chatbot(prompt, max_length=150, num_return_sequences=1)
    return response[0]['generated_text']
user_input = "Give three tips for staying healthy."
response = generate_response(user_input)
print("Bot:", response)