viru / app.py
suprimedev's picture
Update app.py
cd9057b verified
raw
history blame contribute delete
255 Bytes
import gradio as gr
# Use a pipeline as a high-level helper
from transformers import pipeline
messages = [
{"role": "user", "content": "Who are you?"},
]
pipe = pipeline("text-generation", model="meta-llama/Meta-Llama-3.1-8B-Instruct")
pipe(messages)