import streamlit as st
from transformers import pipeline


pipe = pipeline("text-generation", model="Alok64/Llama-2-7b-chat-finetune")
text = st.text_area('enter')

if text:
    out = pipe(text)
    st.json(out)