TeamHaltmannSusanaHWCEO commited on
Commit
3253f49
1 Parent(s): 300a97f

Create D.A.N V3.app

Browse files
Files changed (1) hide show
  1. 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)