llama2-json / app.py
Alok64's picture
Create app.py
f1c5256 verified
raw
history blame contribute delete
211 Bytes
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)