File size: 228 Bytes
dff1849
 
 
 
 
 
 
1
2
3
4
5
6
7
from transformers import pipeline

model_name = "colab-user/distilbert-finetuned-classifier"
chatbot = pipeline("text-generation", model=model_name)

response = chatbot("Hello, how are you?")
print(response[0]['generated_text'])