karthik18AI's picture
Update app.py
9c711cb verified
raw
history blame
240 Bytes
# Use a pipeline as a high-level helper
from transformers import pipeline
messages = [
{"role": "user", "content": "Who are you?"},
]
pipe = pipeline("text-generation", model="Mxode/minicoder-7M-init",max_new_tokens=100)
pipe(messages)