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)