File size: 211 Bytes
f1c5256
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
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)