App / app
colab-user's picture
Create app
dff1849 verified
raw
history blame contribute delete
228 Bytes
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'])