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