math_model / app.py
Aniruddharoy521's picture
app
76f944f verified
raw
history blame contribute delete
No virus
268 Bytes
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)