CSI4160_mHaP / app.py
Katpeeler's picture
text gen concept
172c981
raw
history blame
507 Bytes
import streamlit as st
from transformers import pipeline
from transformers import AutoTokenizer
from transformers import AutoModelForCausalLM
#pipe = pipeline('sentiment-analysis')
pipe = pipeline("text-generation", model="openchat/openchat_3.5")
text = st.text_area('enter some text!')
if text:
out = pipe(text)
st.json(out)
#Reference
#GPT4 Correct User: Hello<|end_of_turn|>GPT4 Correct Assistant: Hi<|end_of_turn|>GPT4 Correct User: How are you today?<|end_of_turn|>GPT4 Correct Assistant: