Aniruddharoy521 commited on
Commit
76f944f
1 Parent(s): 9363778
Files changed (1) hide show
  1. app.py +10 -0
app.py ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ # Use a pipeline as a high-level helper
3
+ from transformers import pipeline
4
+
5
+ pipe = pipeline("text-generation", model="internlm/internlm2-math-plus-mixtral8x22b")
6
+ text = st.text_area("Enter query")
7
+
8
+ if text:
9
+ out = pipe(text)
10
+ st.write(out)