a.py

import streamlit as st
from transformers import pipeline

pipe = pipeline(
    task='text-generation',
    model = 'meta-llama/Meta-Llama-3-8B'
)

x = st.chat_input(placeholder='Your Message')
o = pipe("Hi")
st.write(o[0]['generated_text'])