TeamHaltmannSusanaHWCEO
commited on
Commit
•
3253f49
1
Parent(s):
300a97f
Create D.A.N V3.app
Browse files- D.A.N V3.app +17 -0
D.A.N V3.app
ADDED
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import streamlit as st
|
2 |
+
import gpt_neox
|
3 |
+
|
4 |
+
st.title('GPT-NeoX Streamlit App')
|
5 |
+
|
6 |
+
model = gpt_neox.GPTNeoX(model_name='D.A.N Rev III O.R.CA Edition')
|
7 |
+
|
8 |
+
@st.cache
|
9 |
+
def get_prediction(input_text):
|
10 |
+
return model.predict(input_text)
|
11 |
+
|
12 |
+
st.write('Enter your input text below to get a prediction from the GPT-NeoX model:')
|
13 |
+
user_input = st.text_input('Input Text')
|
14 |
+
|
15 |
+
if user_input:
|
16 |
+
prediction = get_prediction(user_input)
|
17 |
+
st.write('Prediction:', prediction)
|