File size: 268 Bytes
76f944f
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
import streamlit as st
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="internlm/internlm2-math-plus-mixtral8x22b")
text = st.text_area("Enter query")

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